fix forgotten signature change for Short

FluxML · Jun 21, 2021 · 6628c4b · 6628c4b
1 parent 148349e
commit 6628c4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/builders.jl b/src/builders.jl
@@ -46,7 +46,7 @@ mutable struct Short <: Builder
     σ
 end
 Short(; n_hidden=0, dropout=0.5, σ=Flux.sigmoid) = Short(n_hidden, dropout, σ)
-function build(builder::Short, n, m)
+function build(builder::Short, rng, n, m)
     n_hidden =
         builder.n_hidden == 0 ? round(Int, sqrt(n*m)) : builder.n_hidden
     init=Flux.glorot_uniform(rng)