Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
dcuai
dlexamples
Commits
1e6e6a83
Commit
1e6e6a83
authored
Dec 28, 2022
by
sunxx1
Browse files
Merge branch 'pan_dev' into 'main'
添加多机多卡测试方法 See merge request dcutoolkit/deeplearing/dlexamples_new!40
parents
c3e8ee02
d854e7c0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
3 deletions
+16
-3
PyTorch/Compute-Vision/Accuracy_Validation/ResNet50/README.md
...rch/Compute-Vision/Accuracy_Validation/ResNet50/README.md
+16
-3
No files found.
PyTorch/Compute-Vision/Accuracy_Validation/ResNet50/README.md
View file @
1e6e6a83
...
@@ -20,9 +20,7 @@ export HIP_VISIBLE_DEVICES=0
...
@@ -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}/
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
...
@@ -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 --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
```
# 参数说明
# 参数说明
```
```
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment