diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py index 032aa993b0..d79051516f 100644 --- a/deepmd/pt/train/training.py +++ b/deepmd/pt/train/training.py @@ -118,9 +118,6 @@ def __init__( training_params = config["training"] self.multi_task = "model_dict" in model_params self.finetune_links = finetune_links - self.finetune_from_multi_task = model_params.pop( - "finetune_from_multi_task", False - ) # should use pop for next finetune self.model_keys = ( list(model_params["model_dict"]) if self.multi_task else ["Default"] ) diff --git a/deepmd/pt/utils/finetune.py b/deepmd/pt/utils/finetune.py index c4f1abe6bf..e178e856bb 100644 --- a/deepmd/pt/utils/finetune.py +++ b/deepmd/pt/utils/finetune.py @@ -42,7 +42,6 @@ def change_finetune_model_params_single( ) single_config["new_type_map"] = new_type_map else: - single_config["finetune_from_multi_task"] = from_multitask model_dict_params = _model_param_pretrained["model_dict"] new_fitting = False if model_branch_from == "":