Skip to content

Commit

Permalink
delete unused param finetune_from_multi_task
Browse files Browse the repository at this point in the history
  • Loading branch information
iProzd committed Mar 17, 2024
1 parent 3857642 commit 9829bad
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 4 deletions.
3 changes: 0 additions & 3 deletions deepmd/pt/train/training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]
)
Expand Down
1 change: 0 additions & 1 deletion deepmd/pt/utils/finetune.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 == "":
Expand Down

0 comments on commit 9829bad

Please sign in to comment.