How to freeze a multi-task DPA model

[EricGaoOm]

After training the multi-task model, I try to use "dp --pt freeze" to freeze the checkpoint, however, the script told me to add a head. I also tried "dp --pt freeze -o frozen.pth", it just not work.

[njzjz]

Please follow the documentation: https://docs.deepmodeling.com/projects/deepmd/en/latest/freeze/freeze.html

[EricGaoOm]

Many thanks，it works well