ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 00:56:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix launch_multi_node enforce the parent of sub-tasks to be the master node 0 task

Browse Source
This commit is contained in:
allegroai 2024-07-04 15:26:17 +03:00
parent 81b4c49f8b
commit 3fa5409283
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clearml/task.py
View File

@ -2045,7 +2045,7 @@ class Task(_Task):
set_launch_multi_node_runtime_props(self, master_conf) set_launch_multi_node_runtime_props(self, master_conf)
current_conf = master_conf current_conf = master_conf
for node_rank in range(1, master_conf.get("total_num_nodes", total_num_nodes)): for node_rank in range(1, master_conf.get("total_num_nodes", total_num_nodes)):
node = self.clone(source_task=self) node = self.clone(source_task=self, parent=self.id)
node_conf = copy.deepcopy(master_conf) node_conf = copy.deepcopy(master_conf)
node_conf["node_rank"] = node_rank node_conf["node_rank"] = node_rank
set_launch_multi_node_runtime_props(node, node_conf) set_launch_multi_node_runtime_props(node, node_conf)