Update descriptor.py

iProzd · Dec 18, 2024 · ffb70c3 · ffb70c3
1 parent 63040af
commit ffb70c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/model/descriptor/descriptor.py b/deepmd/pt/model/descriptor/descriptor.py
@@ -138,7 +138,7 @@ def share_params(self, base_class, shared_level, resume=False) -> None:
         if shared_level == 0:
             # link buffers
             if hasattr(self, "mean"):
-                if not resume:
+                if not resume or (getattr(self, "skip_stat", False)):
                     # in case of change params during resume
                     base_env = EnvMatStatSe(base_class)
                     base_env.stats = base_class.stats