!4459 modify ssd and yolov3_quant script

Merge pull request !4459 from chengxb7532/master

model_zoo/official/cv/ssd/train.py

@@ -40,7 +40,7 @@ def main():
     parser.add_argument("--lr", type=float, default=0.05, help="Learning rate, default is 0.05.")
     parser.add_argument("--mode", type=str, default="sink", help="Run sink mode or not, default is sink.")
     parser.add_argument("--dataset", type=str, default="coco", help="Dataset, defalut is coco.")
-    parser.add_argument("--epoch_size", type=int, default=250, help="Epoch size, default is 250.")
+    parser.add_argument("--epoch_size", type=int, default=500, help="Epoch size, default is 500.")
     parser.add_argument("--batch_size", type=int, default=32, help="Batch size, default is 32.")
     parser.add_argument("--pre_trained", type=str, default=None, help="Pretrained Checkpoint file path.")
     parser.add_argument("--pre_trained_epoch_size", type=int, default=0, help="Pretrained epoch size.")

model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_eval.sh

@@ -14,9 +14,9 @@
 # limitations under the License.
 # ============================================================================
 
-if [ $# != 2 ]
+if [ $# != 3 ]
 then
-    echo "Usage: sh run_eval.sh [DATASET_PATH] [CHECKPOINT_PATH]"
+    echo "Usage: sh run_eval.sh [DATASET_PATH] [CHECKPOINT_PATH] [DEVICE_ID]"
 exit 1
 fi
 
@@ -45,19 +45,19 @@ exit 1
 fi
 
 export DEVICE_NUM=1
-export DEVICE_ID=0
+export DEVICE_ID=$3
 export RANK_SIZE=$DEVICE_NUM
 export RANK_ID=0
 
-if [ -d "eval" ];
+if [ -d "eval$3" ];
 then
-    rm -rf ./eval
+    rm -rf ./eval$3
 fi
 
-mkdir ./eval
-cp ../*.py ./eval
-cp -r ../src ./eval
-cd ./eval || exit
+mkdir ./eval$3
+cp ../*.py ./eval$3
+cp -r ../src ./eval$3
+cd ./eval$3 || exit
 env > env.log
 echo "start infering for device $DEVICE_ID"
 python eval.py \