diff --git a/model_zoo/official/cv/ssd/train.py b/model_zoo/official/cv/ssd/train.py index c38026103d..73fd3391a1 100644 --- a/model_zoo/official/cv/ssd/train.py +++ b/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.") diff --git a/model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_eval.sh b/model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_eval.sh index ad15d6c6df..89debe747c 100644 --- a/model_zoo/official/cv/yolov3_darknet53_quant/scripts/run_eval.sh +++ b/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 \