From 0a765b1d7de1c336e78cb8f295e326db6f60fcd2 Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Thu, 24 Nov 2022 18:21:09 +0100 Subject: [PATCH] fix test (#2123) --- test/train.jl | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/train.jl b/test/train.jl index cfadde7d9b..f8d66a1e4b 100644 --- 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)