atomic model is not required to provide the fitting net (deepmodeling…

…#3184) Co-authored-by: Han Wang <wang_han@iapcm.ac.cn>
njzjz · Jan 27, 2024 · 2e5333d · 2e5333d
1 parent f4d7c7e
commit 2e5333d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/deepmd/pt/model/model/atomic_model.py b/deepmd/pt/model/model/atomic_model.py
@@ -14,16 +14,9 @@
 from deepmd.model_format import (
     FittingOutputDef,
 )
-from deepmd.pt.model.task import (
-    Fitting,
-)
 
 
 class AtomicModel(ABC):
-    @abstractmethod
-    def get_fitting_net(self) -> Fitting:
-        raise NotImplementedError
-
     @abstractmethod
     def get_fitting_output_def(self) -> FittingOutputDef:
         raise NotImplementedError

diff --git a/deepmd/pt/model/model/dp_atomic_model.py b/deepmd/pt/model/model/dp_atomic_model.py
@@ -128,12 +128,6 @@ def __init__(
                     self.descriptor.dim_out, self.ntypes - 1, self.descriptor.dim_emb
                 )
 
-    def get_fitting_net(self) -> Fitting:
-        """Get the fitting net."""
-        return (
-            self.fitting_net if self.fitting_net is not None else self.coord_denoise_net
-        )
-
     def get_fitting_output_def(self) -> FittingOutputDef:
         """Get the output def of the fitting net."""
         return (