Bug summary

I found a bug when doing single task finetuning(not energy head) based on multitask pre-trained model, the reason I have found.

DeePMD-kit Version

devel branch

TensorFlow Version

not related

How did you download the software?

Offline packages

Input Files, Running Commands, Error Log, etc.

Bug in deepmd/pt/utils/finetune.py change_finetune_model_params lines 138--150. If I do single task finetuning(not energy head) based on multitask pre-trained model, "model_branch" is "", so finetune_links["Default"] is "". So in deepmd/pt/train/training.py line 551, _model_key_from is "". So there is an error:

Steps to Reproduce

Doing single task finetuning based on multitask pre-trained model. Do not specify a model branch.

Further Information, Files, and Links

No response