Skip to content

Commit 934b9db

Browse files
bors[bot]Mottl
andauthored
Merge #1729
1729: Adds affine and track_stats params to BatchNorm docstring r=DhairyaLGandhi a=Mottl ### PR Checklist - [ ] Tests are added - [ ] Entry in NEWS.md - [*] Documentation, if applicable - [ ] API changes require approval from a committer (different from the author, if applicable) Co-authored-by: Dmitry Mottl <dmitry.mottl@gmail.com>
2 parents 327d44d + 2977e98 commit 934b9db

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)