Fix bug in preprocess_shared_params when using non-zero share level (…

…#3615) Signed-off-by: Chenqqian Zhang <[email protected]>
deepmodeling · Mar 27, 2024 · 8090fa1 · 8090fa1
1 parent f5b7aa3
commit 8090fa1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/utils/multi_task.py b/deepmd/pt/utils/multi_task.py
@@ -115,7 +115,7 @@ def replace_one_item(params_dict, key_type, key_in_dict, suffix="", index=None):
                 type_map_keys.append(key_in_dict)
         else:
             if shared_key not in shared_links:
-                class_name = get_class_name(shared_type, shared_dict[key_in_dict])
+                class_name = get_class_name(shared_type, shared_dict[shared_key])
                 shared_links[shared_key] = {"type": class_name, "links": []}
             link_item = {
                 "model_key": model_key,