deepmd.tf.descriptor.se_a_ebd_v2

Attributes

log

Classes

DescrptSeAEbdV2

A compressible se_a_ebd model.

Module Contents

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, 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 wrapper for DescriptorSeA, which set tebd_input_mode=’strip’.