update test

2019-03-14 21:55:37 +08:00 · 2019-03-14 21:55:37 +08:00 · 59da68b4d9
parent 934f0840b2
commit 59da68b4d9
1 changed files with 2 additions and 2 deletions
--- a/test/layers/normalisation.jl
+++ b/test/layers/normalisation.jl
@ -29,10 +29,10 @@ using Flux.Tracker: data

  x = rand(100, 50)
  m = Dropout(0.5)
-  y = m(x, (100, 1))
+  y = m(x, 2)
  c = map(i->count(a->a==0, @view y[:, i]), 1:50)
  @test minimum(c) == maximum(c)
-  y = m(x, (1, 50))
+  y = m(x, 1)
  c = map(i->count(a->a==0, @view y[i, :]), 1:100)
  @test minimum(c) == maximum(c)
 end