We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dd5d110 commit f7bbd5aCopy full SHA for f7bbd5a
src/layers/normalise.jl
@@ -51,6 +51,9 @@ end
51
52
Dropout layer. In the forward pass, apply the [`Flux.dropout`](@ref) function on the input.
53
54
+For N-D dropout layers (e.g. `Dropout2d` or `Dropout3d` in PyTorch),
55
+specify the `dims` keyword (i.e. `Dropout(p; dims = 2)` is a 2D dropout layer).
56
+
57
Does nothing to the input once [`Flux.testmode!`](@ref) is `true`.
58
"""
59
mutable struct Dropout{F,D}
0 commit comments