diff --git a/deepmd/pt/model/descriptor/descriptor.py b/deepmd/pt/model/descriptor/descriptor.py index d97f8964c4..9a61ddca7c 100644 --- 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