p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22952 reset configure of all_reduce_fusion_config for resnet 

Merge pull request !22952 from zhouneng/code_docs_fix_resnet_fusion
This commit is contained in:
i-robot 2021-09-07 03:08:19 +00:00 committed by Gitee
parent 2aedeac6bc af7096db18
commit a1a6c190e1
5 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
model_zoo/official/cv/resnet/config/resnet101_imagenet2012_config.yaml
View File

@ -54,9 +54,9 @@ acc_mode: "O0"
conv_init: "XavierUniform" conv_init: "XavierUniform"
dense_init: "TruncatedNormal" dense_init: "TruncatedNormal"
all_reduce_fusion_config: all_reduce_fusion_config:
- 2 - 80
- 60 - 210
- 220 - 313
train_image_size: 224 train_image_size: 224
eval_image_size: 224 eval_image_size: 224

4
model_zoo/official/cv/resnet/config/resnet50_cifar10_config.yaml
View File

@ -54,8 +54,8 @@ acc_mode: "O0"
conv_init: "XavierUniform" conv_init: "XavierUniform"
dense_init: "TruncatedNormal" dense_init: "TruncatedNormal"
all_reduce_fusion_config: all_reduce_fusion_config:
- 2 - 85
- 115 - 160
train_image_size: 224 train_image_size: 224
eval_image_size: 224 eval_image_size: 224

4
model_zoo/official/cv/resnet/config/se-resnet50_imagenet2012_config.yaml
View File

@ -57,8 +57,8 @@ acc_mode: "O0"
conv_init: "XavierUniform" conv_init: "XavierUniform"
dense_init: "TruncatedNormal" dense_init: "TruncatedNormal"
all_reduce_fusion_config: all_reduce_fusion_config:
- 1 - 85
- 100 - 160
train_image_size: 224 train_image_size: 224
eval_image_size: 256 eval_image_size: 256

2
model_zoo/official/nlp/transformer/README.md
View File

@ -358,7 +358,7 @@ Parameters for learning rate:
### [Export MindIR](#contents) ### [Export MindIR](#contents)
```shell ```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, The ckpt_file parameter is required,

2
model_zoo/official/nlp/transformer/README_CN.md
View File

@ -361,7 +361,7 @@ Parameters for learning rate:
### [导出MindIR](#contents) ### [导出MindIR](#contents)
```shell ```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为必填项， 参数ckpt_file为必填项，