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 94be20f commit 5cac6f6Copy full SHA for 5cac6f6
src/layers/normalise.jl
@@ -51,8 +51,9 @@ end
51
52
Dropout layer. In the forward pass, apply the [`Flux.dropout`](@ref) function on the input.
53
54
-To apply dropout along an certain dimension (e.g. zeroing out an entire channel's feature map),
55
-specify the `dims` keyword (i.e. `Dropout(p; dims = 3)` is a 2D dropout layer on WHCN input).
+To apply dropout along certain dimension(s), specify the `dims` keyword.
+e.g. `Dropout(p; dims = 3)` will randomly zero out entire channels on WHCN input
56
+(also called 2D dropout).
57
58
Does nothing to the input once [`Flux.testmode!`](@ref) is `true`.
59
"""
0 commit comments