[BUG] No share params in self.repinit_three_body in multitask process

[Chengqian-Zhang]

Bug summary

In share_params function of DescrptDPA2 module, repinit_three_body module does not share params.

Code:
if shared_level == 0:
self._modules["type_embedding"] = base_class._modules["type_embedding"]
self.repinit.share_params(base_class.repinit, 0, resume=resume)
self._modules["g1_shape_tranform"] = base_class._modules[
"g1_shape_tranform"
]
self.repformers.share_params(base_class.repformers, 0, resume=resume)

You can see only repinit and repformer share params. repinit_three_body does not.

DeePMD-kit Version

newest devel branch

Backend and its version

not related

How did you download the software?

Offline packages

Input Files, Running Commands, Error Log, etc.

No content

Steps to Reproduce

No content

Further Information, Files, and Links

No content