deepmd.tf.descriptor.se_a_ebd_v2

Module Contents

Classes

DescrptSeAEbdV2

A compressible se_a_ebd model.

Attributes

log

deepmd.tf.descriptor.se_a_ebd_v2.log

class deepmd.tf.descriptor.se_a_ebd_v2.DescrptSeAEbdV2(rcut: float, rcut_smth: float, sel: List[int], neuron: List[int] = [24, 48, 96], axis_neuron: int = 8, resnet_dt: bool = False, trainable: bool = True, seed: int | None = None, type_one_side: bool = True, exclude_types: List[List[int]] = [], set_davg_zero: bool = False, activation_function: str = 'tanh', precision: str = 'default', uniform_seed: bool = False, multi_task: bool = False, spin: deepmd.tf.utils.spin.Spin | None = None, **kwargs)

Bases: deepmd.tf.descriptor.se_a.DescrptSeA

A compressible se_a_ebd model.

This model is a warpper for DescriptorSeA, which set stripped_type_embedding=True.