diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl index 5a8bdc56..728c91df 100644 --- a/src/layers/normalise.jl +++ b/src/layers/normalise.jl @@ -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.