deepmd.dpmodel.fitting.make_base_fitting
Module Contents
Functions
| Make the base class for the fitting. |
- deepmd.dpmodel.fitting.make_base_fitting.make_base_fitting(t_tensor, fwd_method_name: str = 'forward')[source]
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”.