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
ModelZoo
ResNet50_tensorflow
Commits
d67964e2
Commit
d67964e2
authored
Mar 30, 2023
by
qianyj
Browse files
update README
parent
c71d1885
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
README.md
README.md
+4
-1
No files found.
README.md
View file @
d67964e2
...
@@ -43,6 +43,7 @@ python依赖安装:
...
@@ -43,6 +43,7 @@ python依赖安装:
#### 多机多卡训练指令(以单机四卡模拟四卡四进程为例):
#### 多机多卡训练指令(以单机四卡模拟四卡四进程为例):
sed指令只需要执行一次,添加支持多卡运行的代码
sed指令只需要执行一次,添加支持多卡运行的代码
sed -i '100 r configfile' models-master/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
sed -i '100 r configfile' models-master/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
不打开xla:
不打开xla:
...
@@ -83,7 +84,9 @@ sed指令只需要执行一次,添加支持多卡运行的代码
...
@@ -83,7 +84,9 @@ sed指令只需要执行一次,添加支持多卡运行的代码
#### 多机多卡训练指令(以单机四卡模拟四卡四进程为例)
#### 多机多卡训练指令(以单机四卡模拟四卡四进程为例)
sed指令只需要执行一次,添加支持多卡运行的代码
sed指令只需要执行一次,添加支持多卡运行的代码
sed -i '100 r configfile' models-master/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
sed -i '100 r configfile' models-master/official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
修改scripts-run/single_process.sh和scripts-run/single_process_xla.sh文件里的--dtype=fp16
修改scripts-run/single_process.sh和scripts-run/single_process_xla.sh文件里的--dtype=fp16
不打开xla:
不打开xla:
...
@@ -102,7 +105,7 @@ sed指令只需要执行一次,添加支持多卡运行的代码
...
@@ -102,7 +105,7 @@ sed指令只需要执行一次,添加支持多卡运行的代码
根据模型情况填写表格:
根据模型情况填写表格:
| 卡数 | batch size | 类型 | 性能 | Accuracy | 是否打开xla | 进程数 |
| 卡数 | batch size | 类型 | 性能 | Accuracy | 是否打开xla | 进程数 |
| :------: | :------: | :------: | :------: |:------: | :------: |
| :------: | :------: | :------: | :------: |
:------: | :------:
| --------
|
| 4 | 512 | fp32 | 843 examples/second | 0.7628 | 否 | 单进程 |
| 4 | 512 | fp32 | 843 examples/second | 0.7628 | 否 | 单进程 |
| 4 | 512 | fp16 | - | 0.7616 | 否 | 单进程 |
| 4 | 512 | fp16 | - | 0.7616 | 否 | 单进程 |
| 4 | 512 | fp32 | - | 0.7608 | 否 | 四进程 |
| 4 | 512 | fp32 | - | 0.7608 | 否 | 四进程 |
...
...
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