deepmd.jax.descriptor.dpa2

Classes

DescrptDPA2

The unit operation of a native model.

Module Contents

class deepmd.jax.descriptor.dpa2.DescrptDPA2(ntypes: int, repinit: RepinitArgs | dict, repformer: RepformerArgs | dict, concat_output_tebd: bool = True, precision: str = 'float64', smooth: bool = True, exclude_types: list[tuple[int, int]] = [], env_protection: float = 0.0, trainable: bool = True, seed: int | list[int] | None = None, add_tebd_to_repinit_out: bool = False, use_econf_tebd: bool = False, use_tebd_bias: bool = False, type_map: list[str] | None = None)

Bases: deepmd.dpmodel.descriptor.dpa2.DescrptDPA2

The unit operation of a native model.

__setattr__(name: str, value: Any) → None