Skip to content

Commit f8c87d0

Browse files
committed
Adds affine and track_stats params to BatchNorm docstring
1 parent 327d44d commit f8c87d0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/layers/normalise.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,7 @@ end
199199
"""
200200
BatchNorm(channels::Integer, λ=identity;
201201
initβ=zeros32, initγ=ones32,
202+
affine=true, track_stats=true,
202203
ϵ=1f-5, momentum= 0.1f0)
203204
204205
[Batch Normalization](https://arxiv.org/abs/1502.03167) layer.

0 commit comments

Comments
 (0)