Skip to content

Commit

Permalink
pt: fix typo in multitask finetune (#3607)
Browse files Browse the repository at this point in the history
fix #3604 , when doing single-task finetuning from multitask pretrained
model and do not define the finetune model branch from command-line or
input file.
  • Loading branch information
iProzd authored Mar 26, 2024
1 parent 08b3814 commit 3b5b805
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion deepmd/pt/utils/finetune.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,9 @@ def change_finetune_model_params(finetune_model, model_config, model_branch=""):
model_branch_from=model_branch,
)
finetune_links["Default"] = (
model_branch if finetune_from_multi_task else "Default"
model_config["model_branch_chosen"]
if finetune_from_multi_task
else "Default"
)
else:
assert model_branch == "", (
Expand Down

0 comments on commit 3b5b805

Please sign in to comment.