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
65ff3f3e
Commit
65ff3f3e
authored
Feb 03, 2023
by
qianyj
Browse files
update TensorFlow test code
parent
db6b7787
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
16 additions
and
6 deletions
+16
-6
TensorFlow/Accuracy_Validation/README.md
TensorFlow/Accuracy_Validation/README.md
+8
-4
TensorFlow2x/Accuracy_Validation/ResNet50_Official/README.md
TensorFlow2x/Accuracy_Validation/ResNet50_Official/README.md
+5
-1
TensorFlow2x/ComputeVision/Classification/README.md
TensorFlow2x/ComputeVision/Classification/README.md
+3
-1
No files found.
TensorFlow/Accuracy_Validation/README.md
View file @
65ff3f3e
...
@@ -28,11 +28,15 @@
...
@@ -28,11 +28,15 @@
## 运行
## 运行
mpirun -np 4 --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
mpirun -np 4 --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
hostfile格式参考:
node1 slots=4
# 测试说明
node2 slots=4
hostfile格式参考:
node1 slots=4
node2 slots=4
上面多卡测试方法默认每节点4卡
# 参考资料
# 参考资料
[
https://github.com/tensorflow/benchmarks/tree/master/scripts/tf_cnn_benchmarks
](
https://github.com/tensorflow/benchmarks/tree/master/scripts/tf_cnn_benchmarks
)
[
https://github.com/tensorflow/benchmarks/tree/master/scripts/tf_cnn_benchmarks
](
https://github.com/tensorflow/benchmarks/tree/master/scripts/tf_cnn_benchmarks
)
...
...
TensorFlow2x/Accuracy_Validation/ResNet50_Official/README.md
View file @
65ff3f3e
...
@@ -21,6 +21,7 @@
...
@@ -21,6 +21,7 @@
# 多卡测试 (单精度)
# 多卡测试 (单精度)
## 运行
## 运行
# sed指令只需要执行一次,添加支持多卡运行的代码
# sed指令只需要执行一次,添加支持多卡运行的代码
sed -i '99 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
sed -i '99 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
...
@@ -30,6 +31,7 @@
...
@@ -30,6 +31,7 @@
# 多卡测试 (混合精度)
# 多卡测试 (混合精度)
## 运行
## 运行
# sed指令只需要执行一次,添加支持多卡运行的代码
# sed指令只需要执行一次,添加支持多卡运行的代码
sed -i '99 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
sed -i '99 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
...
@@ -37,15 +39,17 @@
...
@@ -37,15 +39,17 @@
export PYTHONPATH=/path/to/tensorflow/model:$PYTHONPATH
export PYTHONPATH=/path/to/tensorflow/model:$PYTHONPATH
mpirun -np $np --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
mpirun -np $np --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
#
# 测试说明
# 测试说明
多卡测试时需要修改部分代码,具体可参考https://tensorflow.google.cn/guide/migrate/multi_worker_cpu_gpu_training?hl=en
多卡测试时需要修改部分代码,具体可参考https://tensorflow.google.cn/guide/migrate/multi_worker_cpu_gpu_training?hl=en
hostfile格式参考:
hostfile格式参考:
node1 slots=4
node1 slots=4
node2 slots=4
node2 slots=4
nodefile格式参考(与hostfile文件内保持一致,用于构造TF_CONFIG):
nodefile格式参考(与hostfile文件内保持一致,用于构造TF_CONFIG):
node1
node1
node2
node2
...
...
TensorFlow2x/ComputeVision/Classification/README.md
View file @
65ff3f3e
# 简介
# 简介
该测试用例用于TensorFlow分类模型性能测试,使用的数据集是imagenet。
该测试用例用于TensorFlow分类模型性能测试,使用的数据集是imagenet。
# 运行
# 运行
...
@@ -29,10 +29,12 @@
...
@@ -29,10 +29,12 @@
多卡测试时需要修改部分代码,具体可参考https://tensorflow.google.cn/guide/migrate/multi_worker_cpu_gpu_training?hl=en
多卡测试时需要修改部分代码,具体可参考https://tensorflow.google.cn/guide/migrate/multi_worker_cpu_gpu_training?hl=en
hostfile格式参考:
hostfile格式参考:
node1 slots=4
node1 slots=4
node2 slots=4
node2 slots=4
nodefile格式参考(与hostfile文件内保持一致,用于构造TF_CONFIG):
nodefile格式参考(与hostfile文件内保持一致,用于构造TF_CONFIG):
node1
node1
node2
node2
...
...
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