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

dropout printing

Browse Source
This commit is contained in:
Mike J Innes 2019-07-24 11:20:39 -04:00
parent faac0ff08b
commit f3551da5a2
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/layers/normalise.jl
View File

@ -37,6 +37,12 @@ end
(a::Dropout)(x) = dropout(x, a.p; dims = a.dims)
function Base.show(io::IO, d::Dropout)
print(io, "Dropout(", d.p)
d.dims != (:) && print(io, ", dims = $(repr(d.dims))")
print(io, ")")
end
"""
AlphaDropout(p)
A dropout layer. It is used in Self-Normalizing Neural Networks.