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

reset all_reduce_fusion_config for resnet

This commit is contained in:
zhouneng2 2021-09-06 11:56:35 +08:00
parent c30ba8ef25
commit af7096db18
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"
dense_init: "TruncatedNormal"
all_reduce_fusion_config:
- 2
- 60
- 220
- 80
- 210
- 313
train_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"
dense_init: "TruncatedNormal"
all_reduce_fusion_config:
- 2
- 115
- 85
- 160
train_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"
dense_init: "TruncatedNormal"
all_reduce_fusion_config:
- 1
- 100
- 85
- 160
train_image_size: 224
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)
```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,

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

@ -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为必填项，