From ffb70c39a39b34c92b95a4597272cf05d7259761 Mon Sep 17 00:00:00 2001
From: Duo <50307526+iProzd@users.noreply.github.com>
Date: Wed, 18 Dec 2024 23:09:35 +0800
Subject: [PATCH] Update descriptor.py

---
 deepmd/pt/model/descriptor/descriptor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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