change fitting to descriptor

Chengqian-Zhang · Sep 23, 2024 · e51a4f7 · e51a4f7
1 parent fc5f1c2
commit e51a4f7
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
@@ -484,11 +484,7 @@ def collect_single_finetune_params(
                             if i != "_extra_state" and f".{_model_key}." in i
                         ]
                         for item_key in target_keys:
-                            if _new_fitting and (
-                                (".fitting_net." in item_key)
-                                or (".out_bias" in item_key)
-                                or (".out_std" in item_key)
-                            ):
+                            if _new_fitting and (".descriptor." not in item_key):
                                 # print(f'Keep {item_key} in old model!')
                                 _new_state_dict[item_key] = (
                                     _random_state_dict[item_key].clone().detach()