fix typo

apache · Nov 29, 2018 · 5281885 · 5281885
1 parent 62dd2d0
commit 5281885
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -1240,10 +1240,10 @@ def test(net, x, record=True):
 
 @with_seed()
 def test_hybrid_static_memory():
-    check_hybrid_static_memory(train_mode=[True, False])
-    check_hybrid_static_memory(train_mode=[True, False], static_alloc=True)
+    check_hybrid_static_memory(train_modes=[True, False])
+    check_hybrid_static_memory(train_modes=[True, False], static_alloc=True)
     # TODO: MKLDNN (issue #13445) does not work with static_shape backwards
-    check_hybrid_static_memory(train_mode=[True], static_alloc=True, static_shape=True)
+    check_hybrid_static_memory(train_modes=[True], static_alloc=True, static_shape=True)
 
 def check_hybrid_static_memory_switching(**kwargs):
     net = gluon.model_zoo.vision.get_resnet(