OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

update_hccl_tools_README.md

This commit is contained in:
GuoMengHao 2020-08-18 10:58:03 +08:00
parent 4394ad846b
commit eef6e49ae4
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
model_zoo/utils/hccl_tools/README.md
View File

@ -5,10 +5,13 @@ mindspore distributed training launch helper utilty that will generate hccl conf
# use
```
python hccl_tools.py --device_num [1,8]
python hccl_tools.py --device_num [0,8)
```
output:
```
hccl_[device_num]p_[which device]_[server_ip].json
```
```
# Note
Please note that the D chips used must be continuous, such [0,4) means to use four chips 0，1，2，3; [0,1) means to use chip 0; The first four chips are a group, and the last four chips are a group. In addition to the [0,8) chips are allowed, other cross-group such as [3,6) are prohibited.

8
model_zoo/utils/hccl_tools/hccl_tools.py
View File

@ -36,12 +36,12 @@ def parse_args():
parser = ArgumentParser(description="mindspore distributed training launch "
"helper utilty that will generate hccl"
" config file")
parser.add_argument("--device_num", type=str, default="[0,8]",
parser.add_argument("--device_num", type=str, default="[0,8)",
help="The number of the D chip used. please note that the D chips"
"used must be continuous, such [0,4] means to use four chips "
"0，1，2，3; [0,1] means to use chip 0; The first four chips are"
"used must be continuous, such [0,4) means to use four chips "
"0，1，2，3; [0,1) means to use chip 0; The first four chips are"
"a group, and the last four chips are a group. In addition to"
"the [0,8] chips are allowed, other cross-group such as [3,6]"
"the [0,8) chips are allowed, other cross-group such as [3,6)"
"are prohibited.")
parser.add_argument("--visible_devices", type=str, default="0,1,2,3,4,5,6,7",
help="will use the visible devices sequentially")