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

Update param descriptions of perator BatchNorm2d

This commit is contained in:
dinglinhe 2021-05-11 10:21:13 +08:00
parent ce57912f7f
commit 0b6597b4a6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/layer/normalization.py
View File

@ -388,7 +388,7 @@ class BatchNorm2d(_BatchNorm):
- If None, The use_batch_statistics is automatically assigned process according to - If None, The use_batch_statistics is automatically assigned process according to
the training and eval mode. During training, batchnorm2d process will be the same the training and eval mode. During training, batchnorm2d process will be the same
with use_batch_statistics=True. Contrarily, in eval, batchnorm2d process will be the same with use_batch_statistics=True. Contrarily, in eval, batchnorm2d process will be the same
with use_batch_statistics=False. with use_batch_statistics=False. Default: None.
data_format (str): The optional value for data format, is 'NHWC' or 'NCHW'. data_format (str): The optional value for data format, is 'NHWC' or 'NCHW'.
Default: 'NCHW'. Default: 'NCHW'.