deepmd.dpmodel.model.base_model

Module Contents

Classes

BaseModel

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

Functions

make_base_model(→ Type[object])

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.

get_model_def_script() → str

Get the model definition script.