Commit 4041a397 authored by panning's avatar panning
Browse files

添加单机四卡训练(混合精度)说明

parent 846c3f70
...@@ -46,12 +46,18 @@ export HIP_VISIBLE_DEVICES=0 ...@@ -46,12 +46,18 @@ export HIP_VISIBLE_DEVICES=0
python3 train.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}/ python3 train.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}/
``` ```
### 单机四卡测试(单精度) ### 单机四卡训练(单精度)
``` ```
mpirun --allow-run-as-root --bind-to none -np 4 scrips/single_process.sh localhost resnet50 64 mpirun --allow-run-as-root --bind-to none -np 4 scrips/single_process.sh localhost resnet50 64
``` ```
### 单机四卡训练(混合精度)
```
mpirun --allow-run-as-root --bind-to none -np 4 scrips/single_process_amp.sh localhost resnet50 64
```
## 准确率数据 ## 准确率数据
测试数据使用ImageNet数据集,使用的加速卡是DCUZ100L。 测试数据使用ImageNet数据集,使用的加速卡是DCUZ100L。
......
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