diff --git a/deepmd/pt/model/task/ener.py b/deepmd/pt/model/task/ener.py index 91ccd03d9a..e40a6bda44 100644 --- a/deepmd/pt/model/task/ener.py +++ b/deepmd/pt/model/task/ener.py @@ -355,7 +355,6 @@ def forward( @Fitting.register("ener") -@fitting_check_output class EnergyFittingNet(InvarFitting): def __init__( self,