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
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=False.
with use_batch_statistics=False. Default: None.
data_format (str): The optional value for data format, is 'NHWC' or 'NCHW'.
Default: 'NCHW'.