From 1996f715dce5ecf2cf3f1be563d9cf8dbebb1d4a Mon Sep 17 00:00:00 2001 From: Payne Date: Fri, 11 Sep 2020 20:52:29 +0800 Subject: [PATCH] add args, close dataset sink mode by default, rename Readme.md -> README.md --- model_zoo/official/cv/mobilenetv2/{Readme.md => README.md} | 2 +- model_zoo/official/cv/mobilenetv2/src/args.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) rename model_zoo/official/cv/mobilenetv2/{Readme.md => README.md} (99%) diff --git a/model_zoo/official/cv/mobilenetv2/Readme.md b/model_zoo/official/cv/mobilenetv2/README.md similarity index 99% rename from model_zoo/official/cv/mobilenetv2/Readme.md rename to model_zoo/official/cv/mobilenetv2/README.md index cc92dde878..f36bed45a3 100644 --- a/model_zoo/official/cv/mobilenetv2/Readme.md +++ b/model_zoo/official/cv/mobilenetv2/README.md @@ -63,7 +63,7 @@ For FP16 operators, if the input data type is FP32, the backend of MindSpore wil ```python ├── MobileNetV2 - ├── Readme.md # descriptions about MobileNetV2 + ├── README.md # descriptions about MobileNetV2 ├── scripts │ ├──run_train.sh # shell script for train, fine_tune or incremental learn with CPU, GPU or Ascend │ ├──run_eval.sh # shell script for evaluation with CPU, GPU or Ascend diff --git a/model_zoo/official/cv/mobilenetv2/src/args.py b/model_zoo/official/cv/mobilenetv2/src/args.py index a72037b301..c92959a359 100644 --- a/model_zoo/official/cv/mobilenetv2/src/args.py +++ b/model_zoo/official/cv/mobilenetv2/src/args.py @@ -60,6 +60,8 @@ def eval_parse_args(): for fine tune or incremental learning') eval_parser.add_argument('--head_ckpt', type=str, default=None, help='Pretrained checkpoint path \ for fine tune or incremental learning') + eval_parser.add_argument('--run_distribute', type=ast.literal_eval, default=False, help='Run distribute') + eval_args = eval_parser.parse_args() return eval_args