fix multitask

Author: iProzd

deepmd/pt/model/descriptor/dpa3.py

@@ -322,6 +322,9 @@ def share_params(self, base_class, shared_level, resume=False) -> None:
         # share all parameters in type_embedding, repflow
         if shared_level == 0:
             self._modules["type_embedding"] = base_class._modules["type_embedding"]
+            for kk in ["chg_embedding", "spin_embedding", "mix_cs_mlp"]:
+                if kk in self._modules:
+                    self._modules[kk] = base_class._modules[kk]
             self.repflows.share_params(base_class.repflows, 0, resume=resume)
         # shared_level: 1
         # share all parameters in type_embedding