diff --git a/official/cv/ResNet/scripts/run_eval.sh b/official/cv/ResNet/scripts/run_eval.sh index 6e3267577d12a7f1bea453dd2f8f85e4de336204..7a9526563e4c2da165a8584b8a883a0826e91c53 100644 --- a/official/cv/ResNet/scripts/run_eval.sh +++ b/official/cv/ResNet/scripts/run_eval.sh @@ -52,7 +52,7 @@ else fi export LOCAL_DEVICE_NUM=1 -export RANK_SIZE=$DEVICE_NUM +export RANK_SIZE=1 export RANK_ID=0 export DEVICE_ID=$DEVICE_ID diff --git a/official/cv/ResNet/scripts/run_eval_gpu.sh b/official/cv/ResNet/scripts/run_eval_gpu.sh index 16ef9bd8766d56e772bce4d94fd67410b3cfc493..2fee2882d6fdf4a8ffdfdc223ae91b2c5975b88c 100644 --- a/official/cv/ResNet/scripts/run_eval_gpu.sh +++ b/official/cv/ResNet/scripts/run_eval_gpu.sh @@ -47,7 +47,7 @@ fi ulimit -u unlimited export LOCAL_DEVICE_NUM=1 -export RANK_SIZE=$DEVICE_NUM +export RANK_SIZE=1 export RANK_ID=0 if [ -d "eval" ]; diff --git a/official/cv/ResNet/scripts/run_eval_onnx_gpu.sh b/official/cv/ResNet/scripts/run_eval_onnx_gpu.sh index 6f5b447d6ab339eaa48c6ca39eac1dafd7ee7ba0..d5f46a1adb70f9812afc14e137b2ff54d585ceb0 100644 --- a/official/cv/ResNet/scripts/run_eval_onnx_gpu.sh +++ b/official/cv/ResNet/scripts/run_eval_onnx_gpu.sh @@ -49,7 +49,7 @@ fi ulimit -u unlimited export LOCAL_DEVICE_NUM=1 -export RANK_SIZE=$DEVICE_NUM +export RANK_SIZE=1 export RANK_ID=0 echo "start evaluation" diff --git a/official/cv/ResNet/scripts/run_infer.sh b/official/cv/ResNet/scripts/run_infer.sh index a14fb51338d66cc903d3f424f20da754b32742f0..4967bdcc7c84fe48be5bf16fcc54da2e9f4c7d03 100644 --- a/official/cv/ResNet/scripts/run_infer.sh +++ b/official/cv/ResNet/scripts/run_infer.sh @@ -47,7 +47,7 @@ fi ulimit -u unlimited export LOCAL_DEVICE_NUM=1 -export RANK_SIZE=$DEVICE_NUM +export RANK_SIZE=1 export RANK_ID=0 if [ -d "infer" ];