diff --git a/test/nets.jl b/test/nets.jl index e7666eea..b06a899a 100644 --- a/test/nets.jl +++ b/test/nets.jl @@ -108,7 +108,7 @@ function test_vae() set_beta!(vae, 0.5, ramp_up=false, steps=100) x,y = first(mb) - p = vae(x) + o = vae(x) return loss isa Real && p isa Dict &&