Skip to content

Commit

Permalink
vae test fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
andreasdominik committed May 12, 2023
1 parent 9c0bbf7 commit 95c35a8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/nets.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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 &&
Expand Down

0 comments on commit 95c35a8

Please sign in to comment.