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

update TensorFlow test code

parent 65ff3f3e
......@@ -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 @@
## 运行
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 @@
sed -i '99 r configfile' official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
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 @@
修改scripts-run/single_process.sh中的--dtype=fp16
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