Skip to content

Commit 1242c20

Browse files
authored
Update normalise.jl
1 parent 185ab40 commit 1242c20

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/layers/normalise.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ The [`Dropout`](@ref) layer is what you should use in most scenarios.
3131
function dropout(x, p; dims=:, active::Bool=true)
3232
active || return x
3333
y = rand!(similar(x, _dropout_shape(x, dims)))
34-
@inbounds @. y = x * _dropout_kernel(y, p, 1-p)
34+
@. y = x * _dropout_kernel(y, p, 1-p)
3535
end
3636

3737
@adjoint function dropout(x, p; dims=:, active::Bool=true)
@@ -56,7 +56,7 @@ e.g. `Dropout(p; dims = 3)` will randomly zero out entire channels on WHCN input
5656
(also called 2D dropout).
5757
5858
Does nothing to the input once [`Flux.testmode!`](@ref) is `true`.
59-
"""`
59+
"""
6060
mutable struct Dropout{F,D}
6161
p::F
6262
dims::D

0 commit comments

Comments
 (0)