Commit 1e6e6a83 authored by sunxx1's avatar sunxx1
Browse files

Merge branch 'pan_dev' into 'main'

添加多机多卡测试方法

See merge request dcutoolkit/deeplearing/dlexamples_new!40
parents c3e8ee02 d854e7c0
......@@ -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