fix test (#2123)

FluxML · Nov 24, 2022 · 0a765b1 · 0a765b1
1 parent a5e5546
commit 0a765b1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/train.jl b/test/train.jl
@@ -46,6 +46,8 @@ end
   end
 
   @testset "non-tuple data" begin
+    w = randn(10, 10)
+    w2 = randn(10, 10)
     loss(m, x) = Flux.Losses.mse(w*x, m.weight*x .+ m.bias)
     model = (weight=copy(w2), bias=zeros(10))
     opt = Flux.setup(AdamW(), model)