File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -60,7 +60,8 @@ class BatchNorm1d(_BatchNorm):
6060 of size N (where N is the input size).
6161
6262 During training, this layer keeps a running estimate of its computed mean
63- and variance. The running sum is kept with a default momentum of 0.1
63+ and variance. The running sum is kept with a default momentum of 0.1.
64+
6465 During evaluation, this running mean/variance is used for normalization.
6566
6667 Args:
@@ -100,7 +101,8 @@ class BatchNorm2d(_BatchNorm):
100101 of size N (where N is the input size).
101102
102103 During training, this layer keeps a running estimate of its computed mean
103- and variance. The running sum is kept with a default momentum of 0.1
104+ and variance. The running sum is kept with a default momentum of 0.1.
105+
104106 During evaluation, this running mean/variance is used for normalization.
105107
106108 Args:
@@ -140,7 +142,8 @@ class BatchNorm3d(_BatchNorm):
140142 of size N (where N is the input size).
141143
142144 During training, this layer keeps a running estimate of its computed mean
143- and variance. The running sum is kept with a default momentum of 0.1
145+ and variance. The running sum is kept with a default momentum of 0.1.
146+
144147 During evaluation, this running mean/variance is used for normalization.
145148
146149 Args:
You can’t perform that action at this time.
0 commit comments