From 3ab4db4adf2d33238fe637b8898c56191be5d3dc Mon Sep 17 00:00:00 2001 From: zhaoting Date: Tue, 30 Jul 2024 15:52:06 +0800 Subject: [PATCH] fix a bug of resnet eval shell --- official/cv/ResNet/scripts/run_eval.sh | 2 +- official/cv/ResNet/scripts/run_eval_gpu.sh | 2 +- official/cv/ResNet/scripts/run_eval_onnx_gpu.sh | 2 +- official/cv/ResNet/scripts/run_infer.sh | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/official/cv/ResNet/scripts/run_eval.sh b/official/cv/ResNet/scripts/run_eval.sh index 6e3267577..7a9526563 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 16ef9bd87..2fee2882d 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 6f5b447d6..d5f46a1ad 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 a14fb5133..4967bdcc7 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" ]; -- Gitee