diff --git a/model_zoo/official/cv/yolov3_resnet18/scripts/run_distribute_train.sh b/model_zoo/official/cv/yolov3_resnet18/scripts/run_distribute_train.sh
index 457598c9d9942140afc9c66a4e2cc350602efe2c..10643e7a6a6c0eeb41871c32ef94e33fcf4d4241 100644
--- a/model_zoo/official/cv/yolov3_resnet18/scripts/run_distribute_train.sh
+++ b/model_zoo/official/cv/yolov3_resnet18/scripts/run_distribute_train.sh
@@ -36,18 +36,22 @@ ANNO_PATH=$5
 PRE_TRAINED=$7
 PRE_TRAINED_EPOCH_SIZE=$8
 
+BASE_PATH=$(cd "`dirname $0`" || exit; pwd)
+cd $BASE_PATH/../ || exit
+
 # Before start distribute train, first create mindrecord files.
 python train.py --only_create_dataset=1 --mindrecord_dir=$MINDRECORD_DIR --image_dir=$IMAGE_DIR  \
 --anno_path=$ANNO_PATH
+if [ $? -ne 0 ]
+then
+    exit 1
+fi
 
 echo "After running the scipt, the network runs in the background. The log will be generated in LOGx/log.txt"
 
 export RANK_TABLE_FILE=$6
 export RANK_SIZE=$1
 
-BASE_PATH=$(cd "`dirname $0`" || exit; pwd)
-cd $BASE_PATH/../ || exit
-
 for((i=0;i<RANK_SIZE;i++))
 do
     export DEVICE_ID=$i
diff --git a/model_zoo/official/cv/yolov3_resnet18/train.py b/model_zoo/official/cv/yolov3_resnet18/train.py
index 761a94c454a151a8b523f60475ab1b8182cd62a3..e0d309be9cd23b467e3205874012c2538582409d 100644
--- a/model_zoo/official/cv/yolov3_resnet18/train.py
+++ b/model_zoo/official/cv/yolov3_resnet18/train.py
@@ -115,7 +115,8 @@ def main():
                                           8)
             print("Create Mindrecord Done, at {}".format(args_opt.mindrecord_dir))
         else:
-            print("image_dir or anno_path not exits.")
+            raise ValueError('image_dir {} or anno_path {} does not exist'.format(\
+                              args_opt.image_dir, args_opt.anno_path))
 
     if not args_opt.only_create_dataset:
         loss_scale = float(args_opt.loss_scale)