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

3
src/layers/normalise.jl
View File

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