diff --git a/model_zoo/official/cv/resnet/config/resnet101_imagenet2012_config.yaml b/model_zoo/official/cv/resnet/config/resnet101_imagenet2012_config.yaml index 738c831b9ad..49d250cda15 100644 --- a/model_zoo/official/cv/resnet/config/resnet101_imagenet2012_config.yaml +++ b/model_zoo/official/cv/resnet/config/resnet101_imagenet2012_config.yaml @@ -54,9 +54,9 @@ acc_mode: "O0" conv_init: "XavierUniform" dense_init: "TruncatedNormal" all_reduce_fusion_config: - - 2 - - 60 - - 220 + - 80 + - 210 + - 313 train_image_size: 224 eval_image_size: 224 diff --git a/model_zoo/official/cv/resnet/config/resnet50_cifar10_config.yaml b/model_zoo/official/cv/resnet/config/resnet50_cifar10_config.yaml index 78cd6ab52b8..d6feb0d326e 100644 --- a/model_zoo/official/cv/resnet/config/resnet50_cifar10_config.yaml +++ b/model_zoo/official/cv/resnet/config/resnet50_cifar10_config.yaml @@ -54,8 +54,8 @@ acc_mode: "O0" conv_init: "XavierUniform" dense_init: "TruncatedNormal" all_reduce_fusion_config: - - 2 - - 115 + - 85 + - 160 train_image_size: 224 eval_image_size: 224 diff --git a/model_zoo/official/cv/resnet/config/se-resnet50_imagenet2012_config.yaml b/model_zoo/official/cv/resnet/config/se-resnet50_imagenet2012_config.yaml index 9bafed98657..7031d024259 100644 --- a/model_zoo/official/cv/resnet/config/se-resnet50_imagenet2012_config.yaml +++ b/model_zoo/official/cv/resnet/config/se-resnet50_imagenet2012_config.yaml @@ -57,8 +57,8 @@ acc_mode: "O0" conv_init: "XavierUniform" dense_init: "TruncatedNormal" all_reduce_fusion_config: - - 1 - - 100 + - 85 + - 160 train_image_size: 224 eval_image_size: 256 diff --git a/model_zoo/official/nlp/transformer/README.md b/model_zoo/official/nlp/transformer/README.md index 3a4ad80b827..5ecc158deff 100644 --- a/model_zoo/official/nlp/transformer/README.md +++ b/model_zoo/official/nlp/transformer/README.md @@ -358,7 +358,7 @@ Parameters for learning rate: ### [Export MindIR](#contents) ```shell -python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] +python export.py --model_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] ``` The ckpt_file parameter is required, diff --git a/model_zoo/official/nlp/transformer/README_CN.md b/model_zoo/official/nlp/transformer/README_CN.md index f14c8435c56..44bcf179a7e 100644 --- a/model_zoo/official/nlp/transformer/README_CN.md +++ b/model_zoo/official/nlp/transformer/README_CN.md @@ -361,7 +361,7 @@ Parameters for learning rate: ### [导出MindIR](#contents) ```shell -python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] +python export.py --model_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] ``` 参数ckpt_file为必填项,