harder test

2018-03-01 16:37:52 +00:00 · 2018-03-01 16:37:52 +00:00 · cf7dd34767
parent 8019f789f8
commit cf7dd34767
1 changed files with 1 additions and 1 deletions
--- a/test/cuda/cuda.jl
+++ b/test/cuda/cuda.jl
@ -15,7 +15,7 @@ x = Flux.onehotbatch([1, 2, 3], 1:3)
 cx = gpu(x)
@test cx isa Flux.OneHotMatrix && cx.data isa CuArray

-m = Chain(Dense(10, 5, σ), Dense(5, 2))
+m = Chain(Dense(10, 5, tanh), Dense(5, 2), softmax)
 cm = gpu(m)

@test all(p isa TrackedArray && p.data isa CuArray for p in params(cm))