Fix docstrings

FluxML · Jan 27, 2022 · f922c16 · f922c16
1 parent 15ac6cd
commit f922c16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl
@@ -55,7 +55,7 @@ function _dropout_mask(rng, x, p; dims=:)
 end
 
 """
-    Dropout(p; dims=:, rng = default_rng())
+    Dropout(p; dims=:, rng = rng_from_array())
 
 Dropout layer. In the forward pass, apply the [`Flux.dropout`](@ref) function on the input.
 
@@ -100,7 +100,7 @@ function Base.show(io::IO, d::Dropout)
 end
 
 """
-    AlphaDropout(p; rng = default_rng())
+    AlphaDropout(p; rng = rng_from_array())
 
 A dropout layer. Used in
 [Self-Normalizing Neural Networks](https://arxiv.org/abs/1706.02515).