Skip to content

Commit 6d14ef8

Browse files
ronrestapaszke
authored andcommitted
Update batchnorm docstrings
Add missing full stops, and added blank line for increased clarity on rendered documentation.
1 parent 26a492a commit 6d14ef8

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

torch/nn/modules/batchnorm.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)