This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
fix globalbatchnorm bug
tags/v0.3.0-alpha
zhaojichen
5 years ago
parent
8261cfd019
commit
6c9a54afa1
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
mindspore/nn/layer/normalization.py
+ 1
- 1
mindspore/nn/layer/normalization.py
View File
@@ -119,7 +119,7 @@ class _BatchNorm(Cell):
def _shape_infer(self, x):
"""global batch normalization shape and axes infer"""
if len(self.shape(x)) == 4:
axes = (0,2,3)
axes = (0,
2,
3)
re_shape = (1, self.num_features, 1, 1)
else:
axes = (0,)
Write
Preview
Loading…
Cancel
Save