From fc6f5bb199b926330d69aad044e094284e54b5fe Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Thu, 24 Nov 2022 16:52:10 +0100 Subject: [PATCH] fix test --- 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)