"src/git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "cf88652aa84b06b8a7c72f40ba91aa998aa7c93a"
Commit 3f2973d6 authored by qianyj's avatar qianyj
Browse files

update TensorFlow test code

parent 65ff3f3e
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
## 运行 ## 运行
mpirun -np 4 --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh mpirun -np ${num_gpu} --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
# 多卡测试 (混合精度) # 多卡测试 (混合精度)
...@@ -27,7 +27,7 @@ ...@@ -27,7 +27,7 @@
## 运行 ## 运行
mpirun -np 4 --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh mpirun -np ${num_gpu} --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
# 测试说明 # 测试说明
......
...@@ -26,7 +26,7 @@ ...@@ -26,7 +26,7 @@
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
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 ${num_gpu} --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
# 多卡测试 (混合精度) # 多卡测试 (混合精度)
...@@ -37,7 +37,7 @@ ...@@ -37,7 +37,7 @@
修改scripts-run/single_process.sh中的--dtype=fp16 修改scripts-run/single_process.sh中的--dtype=fp16
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 ${num_gpu} --hostfile hostfile -mca btl self,tcp --bind-to none scripts-run/single_process.sh
# 测试说明 # 测试说明
......
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