TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Minor update

Browse Source
This commit is contained in:
thebhatman 2019-06-13 18:59:03 +05:30
parent ce6a1bf84f
commit 7ab9d8ed3d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/normalise.jl
View File

@ -29,7 +29,7 @@ _dropout_kernel(y::T, p, q) where {T} = y > p ? T(1 / q) : T(0)
function dropout(x, p; dims = :) function dropout(x, p; dims = :)
istraining() || return x istraining() || return x
y = similar(x) y = similar(x, _dropout_shape(x, dims))
rand!(y) rand!(y)
y .= _dropout_kernel.(y, p, 1 - p) y .= _dropout_kernel.(y, p, 1 - p)
return x .* y return x .* y