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

Silly error in Dropout corrected.

Browse Source
This commit is contained in:
thebhatman 2019-06-12 23:16:15 +05:30
parent e9797408ec
commit 48ed93cdaa
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/layers/normalise.jl
View File

@ -15,10 +15,11 @@ Does nothing to the input once in [`testmode!`](@ref).
mutable struct Dropout{F}
p::F
dims::Union{Colon, Int, NTuple{N, Int} where N}
function Dropout(p; dims = :)
end
function Dropout(p; dims = :)
@assert 0 ≤ p ≤ 1
Dropout{typeof(p)}(p, dims)
end
end
_dropout_shape(s, ::Colon) = size(s)