Commit d854e7c0 authored by panning's avatar panning
Browse files

添加多机多卡测试方法

parent c056df78
......@@ -20,9 +20,7 @@ export HIP_VISIBLE_DEVICES=0
python3 main_acc.py --batch-size=64 --arch=resnet50 -j 6 --epochs=90 --amp --opt-level O1 --loss-scale=dynamic --save-path=/path/to/{save_model_dir} /path/to/{ImageNet_pytorch_data_dir}/
```
# 多卡测试(单精度)
# 单机多卡测试(单精度)
## 运行
......@@ -31,6 +29,21 @@ python3 main_acc.py --batch-size=64 --arch=resnet50 -j 6 --epochs=90 --amp --opt
mpirun --allow-run-as-root --bind-to none -np 4 single_process.sh localhost resnet50 64
```
# 多机多卡测试(单精度)
## 运行
```
mpirun -np ${num_dcu} --hostfile hostfile --allow-run-as-root --bind-to none `pwd`/single_process.sh [node1] resnet50 64
```
hostfile格式参考:
```
node1 slots=4
node2 slots=4
```
# 参数说明
```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment