back on mse failing for Float16

2019-06-13 16:41:25 +05:30 · 2019-06-13 16:41:25 +05:30 · 1ff4e3188e
parent ce11804dc1
commit 1ff4e3188e
1 changed files with 1 additions and 1 deletions
--- a/test/layers/stateless.jl
+++ b/test/layers/stateless.jl
@ -56,7 +56,7 @@ const ϵ = 1e-7
      y = rand(T, 2)
      ŷ = rand(T, 2)
      for f in (mse, crossentropy, logitcrossentropy)
-        fwd, back = Zygote.forward(mse, ŷ, y)
+        fwd, back = Zygote.forward(f, ŷ, y)
        @test fwd isa T
        @test eltype(back(one(T))[1]) == T
      end