Skip to content

Commit af217d5

Browse files
authored
Merge pull request #520 from dingzhaohan/devel
modify decide_machine for older ALI-Dispatcher
2 parents 5682e98 + 9d4396d commit af217d5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

dpgen/remote/decide_machine.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ def convert_mdata(mdata, task_types=["train", "model_devi", "fp"]):
4040
if "comments" not in key:
4141
mdata[task_type + "_" + key] = item
4242
group_size = mdata[task_type][0]["resources"].get("group_size", 1)
43+
if group_size == 1: group_size = mdata[task_type][0].get("group_size", 1)
4344
mdata[task_type + "_" + "group_size"] = group_size
4445
return mdata
4546

0 commit comments

Comments
 (0)