Skip to content

Commit

Permalink
Merge pull request torch#619 from temerick/patch-2
Browse files Browse the repository at this point in the history
Update SpatialBatchNormalization.lua
  • Loading branch information
soumith committed Feb 8, 2016
2 parents 39723b8 + b61a0da commit bf6e425
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions SpatialBatchNormalization.lua
Original file line number Diff line number Diff line change
Expand Up @@ -16,15 +16,15 @@
The learning of gamma and beta is optional.
Usage:
with learnable parameters: nn.BatchNormalization(N [,eps] [,momentum])
with learnable parameters: nn.SpatialBatchNormalization(N [,eps] [,momentum])
where N = dimensionality of input
without learnable parameters: nn.BatchNormalization(N [,eps] [,momentum], false)
without learnable parameters: nn.SpatialBatchNormalization(N [,eps] [,momentum], false)
eps is a small value added to the variance to avoid divide-by-zero.
Defaults to 1e-5
In training time, this layer keeps a running estimate of it's computed mean and std.
The running sum is kept with a default momentup of 0.1 (unless over-ridden)
The running sum is kept with a default momentum of 0.1 (unless over-ridden)
In test time, this running mean/std is used to normalize.
]]--
Expand Down

0 comments on commit bf6e425

Please sign in to comment.