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 f7bbd5a commit ad83666Copy full SHA for ad83666
src/layers/normalise.jl
@@ -52,7 +52,7 @@ end
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).
+specify the `dims` keyword (i.e. `Dropout(p; dims = 3)` is a 2D dropout layer).
56
57
Does nothing to the input once [`Flux.testmode!`](@ref) is `true`.
58
"""
0 commit comments