deepmd.dpmodel.fitting.make_base_fitting

Functions

make_base_fitting(t_tensor[, fwd_method_name])

Make the base class for the fitting.

Module Contents

deepmd.dpmodel.fitting.make_base_fitting.make_base_fitting(t_tensor, fwd_method_name: str = 'forward')

Make the base class for the fitting.

Parameters:

t_tensor

The type of the tensor. used in the type hint.

fwd_method_name

Name of the forward method. For dpmodels, it should be “call”. For torch models, it should be “forward”.