From 29e466f43740505313b67db324d7acaa9b748b9a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20M=C3=A9tivier?= <46794064+dmetivie@users.noreply.github.com> Date: Tue, 16 Jul 2024 04:38:43 +0200 Subject: [PATCH] docs: update docstring dropout (#770) --- src/layers/dropout.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/dropout.jl b/src/layers/dropout.jl index eb84a8a4d..e79983b99 100644 --- a/src/layers/dropout.jl +++ b/src/layers/dropout.jl @@ -71,7 +71,7 @@ Dropout layer. ## Keyword Arguments - 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 + `Dropout(p; dims = (3,4))` will randomly zero out entire channels on WHCN input (also called 2D dropout). ## Inputs