deepmd.dpmodel.model.base_model

Classes

BaseModel

Base class for final exported model that will be directly used for inference.

Functions

make_base_model(→ type[object])

Module Contents

deepmd.dpmodel.model.base_model.make_base_model() → type[object]

class deepmd.dpmodel.model.base_model.BaseModel

Bases: make_base_model()

Base class for final exported model that will be directly used for inference.

The class defines some abstractmethods that will be directly called by the inference interface. If the final model class inherbits some of those methods from other classes, BaseModel should be inherited as the last class to ensure the correct method resolution order.

This class is for the DPModel backend.

See also

deepmd.dpmodel.model.base_model.BaseBaseModel

Backend-independent BaseModel class.

model_def_script = ''

The model definition script.

min_nbor_dist = None

The minimum distance between two atoms. Used for model compression. None when skipping neighbor statistics.

get_model_def_script() → str

Get the model definition script.