Fix for Batchnorm odd & even chnls number context

apache · Oct 29, 2020 · 2c11470 · 2c11470
1 parent 62b58d8
commit 2c11470
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -2210,7 +2210,7 @@ def hybrid_forward(self, F, x):
     for i in range(len(chn_list)):
         net = Net(chn_list[i])
         net.initialize(init=init.Constant(1))
-        x = mx.nd.zeros((1, 3, 8, 160, 160), ctx=mx.cpu())
+        x = mx.nd.zeros((1, 3, 8, 160, 160))
         net(x).asnumpy()