fix dropout

apache · Oct 4, 2019 · f31f982 · f31f982
1 parent 6f7cf46
commit f31f982
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -742,7 +742,8 @@ def test_dropout():
     exe = y.simple_bind(ctx=default_context(), data=shape)
     exe.arg_arrays[0][:] = 1
     out = exe.forward(is_train=True)
-    assert out.shape == out.shape
+    nd.waitall()
+    assert out[0].shape == shape
 
 
 def test_activation():