deepmd.pt package
Subpackages
- deepmd.pt.entrypoints package
- deepmd.pt.infer package
- deepmd.pt.loss package
- deepmd.pt.model package
- Subpackages
- deepmd.pt.model.atomic_model package
BaseAtomicModel
DPAtomicModel
DPZBLLinearAtomicModel
LinearAtomicModel
PairTabAtomicModel
- Submodules
- deepmd.pt.model.atomic_model.base_atomic_model module
- deepmd.pt.model.atomic_model.dp_atomic_model module
- deepmd.pt.model.atomic_model.linear_atomic_model module
- deepmd.pt.model.atomic_model.pairtab_atomic_model module
- deepmd.pt.model.backbone package
- deepmd.pt.model.descriptor package
BaseDescriptor
DescriptorBlock
DescrptBlockHybrid
DescrptBlockRepformers
DescrptBlockSeA
DescrptBlockSeAtten
DescrptDPA1
DescrptDPA2
DescrptGaussianLcc
DescrptHybrid
DescrptSeA
DescrptSeR
make_default_type_embedding()
prod_env_mat()
- Submodules
- deepmd.pt.model.descriptor.base_descriptor module
- deepmd.pt.model.descriptor.descriptor module
- deepmd.pt.model.descriptor.dpa1 module
- deepmd.pt.model.descriptor.dpa2 module
- deepmd.pt.model.descriptor.env_mat module
- deepmd.pt.model.descriptor.gaussian_lcc module
- deepmd.pt.model.descriptor.hybrid module
- deepmd.pt.model.descriptor.repformer_layer module
- deepmd.pt.model.descriptor.repformers module
- deepmd.pt.model.descriptor.se_a module
- deepmd.pt.model.descriptor.se_atten module
- deepmd.pt.model.descriptor.se_r module
- deepmd.pt.model.model package
BaseModel
DPModel
DPZBLModel
EnergyModel
get_model()
get_zbl_model()
make_hessian_model()
make_model()
- Submodules
- deepmd.pt.model.model.dipole_model module
- deepmd.pt.model.model.dp_model module
- deepmd.pt.model.model.dp_zbl_model module
- deepmd.pt.model.model.ener_model module
- deepmd.pt.model.model.make_hessian_model module
- deepmd.pt.model.model.make_model module
- deepmd.pt.model.model.model module
- deepmd.pt.model.model.polar_model module
- deepmd.pt.model.model.transform_output module
- deepmd.pt.model.network package
- deepmd.pt.model.task package
BaseFitting
DenoiseNet
DipoleFittingNet
EnergyFittingNet
EnergyFittingNetDirect
Fitting
FittingNetAttenLcc
PolarFittingNet
TypePredictNet
- Submodules
- deepmd.pt.model.task.atten_lcc module
- deepmd.pt.model.task.base_fitting module
- deepmd.pt.model.task.denoise module
- deepmd.pt.model.task.dipole module
- deepmd.pt.model.task.ener module
- deepmd.pt.model.task.fitting module
- deepmd.pt.model.task.polarizability module
- deepmd.pt.model.task.task module
- deepmd.pt.model.task.type_predict module
- deepmd.pt.model.atomic_model package
- Subpackages
- deepmd.pt.optimizer package
- deepmd.pt.train package
- deepmd.pt.utils package
AtomExcludeMask
PairExcludeMask
- Submodules
- deepmd.pt.utils.ase_calc module
- deepmd.pt.utils.auto_batch_size module
- deepmd.pt.utils.cache module
- deepmd.pt.utils.dataloader module
- deepmd.pt.utils.dataset module
- deepmd.pt.utils.dp_random module
- deepmd.pt.utils.env module
- deepmd.pt.utils.env_mat_stat module
- deepmd.pt.utils.exclude_mask module
- deepmd.pt.utils.finetune module
- deepmd.pt.utils.learning_rate module
- deepmd.pt.utils.multi_task module
- deepmd.pt.utils.neighbor_stat module
- deepmd.pt.utils.nlist module
- deepmd.pt.utils.plugin module
- deepmd.pt.utils.preprocess module
- deepmd.pt.utils.region module
- deepmd.pt.utils.serialization module
- deepmd.pt.utils.stat module
- deepmd.pt.utils.update_sel module
- deepmd.pt.utils.utils module