forked from mindspore-Ecosystem/mindspore
!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:
commit
a1a6c190e1
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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为必填项,
|
||||
|
|
Loading…
Reference in New Issue