!9367 remove redundant parameters of resnext50 and inceptionv3

From: @zhouyaqiang0
Reviewed-by: 
Signed-off-by:

model_zoo/official/cv/inceptionv3/README.md

@@ -90,8 +90,6 @@ For FP16 operators, if the input data type is FP32, the backend of MindSpore wil
 ```python
 Major parameters in train.py and config.py are:
 'random_seed'                # fix random seed
-'rank'                       # local rank of distributed
-'group_size'                 # world size of distributed
 'work_nums'                  # number of workers to read the data
 'decay_method'               # learning rate scheduler mode
 "loss_scale"                 # loss scale

model_zoo/official/cv/inceptionv3/src/config.py

@@ -20,8 +20,6 @@ from easydict import EasyDict as edict
 
 config_gpu = edict({
     'random_seed': 1,
-    'rank': 0,
-    'group_size': 1,
     'work_nums': 8,
     'decay_method': 'cosine',
     "loss_scale": 1,
@@ -47,8 +45,6 @@ config_gpu = edict({
 
 config_ascend = edict({
     'random_seed': 1,
-    'rank': 0,
-    'group_size': 1,
     'work_nums': 8,
     'decay_method': 'cosine',
     "loss_scale": 1024,

model_zoo/official/cv/resnext50/README.md

@@ -32,7 +32,6 @@ The overall network architecture of ResNeXt is show below:
 
 [Link](https://arxiv.org/abs/1611.05431)
 
-
 # [Dataset](#contents)
 
 Dataset used: [imagenet](http://www.image-net.org/)
@@ -40,10 +39,9 @@ Dataset used: [imagenet](http://www.image-net.org/)
 - Dataset size: ~125G, 1.2W colorful images in 1000 classes
 - Train: 120G, 1.2W images
 - Test: 5G, 50000 images
-- Data format: RGB images.
+- Data format: RGB images
 - Note: Data will be processed in src/dataset.py
 
-
 # [Features](#contents)
 
 ## [Mixed Precision](#contents)
@@ -107,7 +105,7 @@ For FP16 operators, if the input data type is FP32, the backend of MindSpore wil
 
 Parameters for both training and evaluating can be set in config.py.
 
-```       
+```config
 "image_height": '224,224'                 # image size
 "num_classes": 1000,                      # dataset class number
 "per_batch_size": 128,                    # batch size of input tensor
@@ -118,7 +116,6 @@ Parameters for both training and evaluating can be set in config.py.
 "eta_min": 0,                             # eta_min in cosine_annealing scheduler
 "T_max": 150,                             # T-max in cosine_annealing scheduler
 "max_epoch": 150,                         # max epoch num to train the model
-"backbone": 'resnext50',                  # backbone metwork
 "warmup_epochs" : 1,                      # warmup epoch
 "weight_decay": 0.0001,                   # weight decay
 "momentum": 0.9,                          # momentum
@@ -135,17 +132,17 @@ Parameters for both training and evaluating can be set in config.py.
 
 ## [Training Process](#contents)
 
-#### Usage
+### Usage
 
 You can start training by python script:
 
-```
+```script
 python train.py --data_dir ~/imagenet/train/ --platform Ascend --is_distributed 0
 ```
 
 or shell stript:
 
-```
+```script
 Ascend:
     # distribute training example(8p)
     sh run_distribute_train.sh RANK_TABLE_FILE DATA_PATH
@@ -180,16 +177,17 @@ You can find checkpoint file together with result in log.
 
 You can start training by python script:
 
-```
+```script
 python eval.py --data_dir ~/imagenet/val/ --platform Ascend --pretrained resnext.ckpt
 ```
 
 or shell stript:
 
-```
+```script
 # Evaluation
 sh run_eval.sh DEVICE_ID DATA_PATH PRETRAINED_CKPT_PATH PLATFORM
 ```
+
 PLATFORM is Ascend or GPU, default is Ascend.
 
 #### Launch
@@ -203,7 +201,7 @@ sh scripts/run_eval.sh 0 /opt/npu/datasets/classification/val /resnext50_100.ckp
 
 Evaluation result will be stored in the scripts path. Under this, you can find result like the followings in log.
 
-```
+```log
 acc=78.16%(TOP1)
 acc=93.88%(TOP5)
 ```
@@ -212,7 +210,7 @@ acc=93.88%(TOP5)
 
 Change the export mode and export file in `src/config.py`, and run `export.py`.
 
-```
+```script
 python export.py --platform PLATFORM --pretrained CKPT_PATH
 ```
 
@@ -248,7 +246,6 @@ python export.py --platform PLATFORM --pretrained CKPT_PATH
 | outputs                    | probability                   | probability               | probability          |
 | Accuracy                   | acc=78.16%(TOP1)              | acc=78.05%(TOP1)          |                      |
 
-
 # [Description of Random Situation](#contents)
 
 In dataset.py, we set the seed inside “create_dataset" function. We also use random seed in train.py.

model_zoo/official/cv/resnext50/eval.py

@@ -74,7 +74,6 @@ def parse_args(cloud_args=None):
     args = merge_args(args, cloud_args)
     args.image_size = config.image_size
     args.num_classes = config.num_classes
-    args.backbone = config.backbone
     args.rank = config.rank
     args.group_size = config.group_size
 
@@ -201,9 +200,7 @@ def test(cloud_args=None):
                                             max_epoch=1, rank=args.rank, group_size=args.group_size,
                                             mode='eval')
         eval_dataloader = de_dataset.create_tuple_iterator(output_numpy=True, num_epochs=1)
-        network = get_network(args.backbone, num_classes=args.num_classes, platform=args.platform)
-        if network is None:
-            raise NotImplementedError('not implement {}'.format(args.backbone))
+        network = get_network(num_classes=args.num_classes, platform=args.platform)
 
         load_pretrain_model(model, network, args)
 

model_zoo/official/cv/resnext50/export.py

@@ -33,7 +33,6 @@ def parse_args():
     args, _ = parser.parse_known_args()
     args.image_size = config.image_size
     args.num_classes = config.num_classes
-    args.backbone = config.backbone
 
     args.image_size = list(map(int, config.image_size.split(',')))
     args.image_height = args.image_size[0]
@@ -46,7 +45,7 @@ if __name__ == '__main__':
     args_export = parse_args()
     context.set_context(mode=context.GRAPH_MODE, device_target=args_export.platform)
 
-    net = get_network(args_export.backbone, num_classes=args_export.num_classes, platform=args_export.platform)
+    net = get_network(num_classes=args_export.num_classes, platform=args_export.platform)
 
     param_dict = load_checkpoint(args_export.pretrained)
     load_param_into_net(net, param_dict)

model_zoo/official/cv/resnext50/src/config.py

@@ -26,7 +26,6 @@ config = ed({
     "eta_min": 0,
     "T_max": 150,
     "max_epoch": 150,
-    "backbone": 'resnext50',
     "warmup_epochs": 1,
 
     "weight_decay": 0.0001,

model_zoo/official/cv/resnext50/src/image_classification.py

@@ -94,7 +94,5 @@ class Resnet(ImageClassificationNetwork):
 
 
 
-def get_network(backbone_name, **kwargs):
-    if backbone_name in ['resnext50']:
-        return Resnet(backbone_name, **kwargs)
-    return None
+def get_network(**kwargs):
+    return Resnet('resnext50', **kwargs)

model_zoo/official/cv/resnext50/train.py

@@ -131,7 +131,6 @@ def parse_args(cloud_args=None):
     args.eta_min = config.eta_min
     args.T_max = config.T_max
     args.max_epoch = config.max_epoch
-    args.backbone = config.backbone
     args.warmup_epochs = config.warmup_epochs
     args.weight_decay = config.weight_decay
     args.momentum = config.momentum
@@ -213,9 +212,7 @@ def train(cloud_args=None):
     # network
     args.logger.important_info('start create network')
     # get network and init
-    network = get_network(args.backbone, num_classes=args.num_classes, platform=args.platform)
-    if network is None:
-        raise NotImplementedError('not implement {}'.format(args.backbone))
+    network = get_network(num_classes=args.num_classes, platform=args.platform)
 
     load_pretrain_model(args.pretrained, network, args)