Fix #11353 (#11360)

* Update test_gluon_trainer.py * Update test_gluon_trainer.py * Update test_gluon_trainer.py * Update test_gluon_trainer.py * Update test_gluon_trainer.py * trigger * Run 100000 times * Update test_gluon_trainer.py * run 10K times * test_trainer_reset_kv didn't fail for 10K time . 2nd Trigger. * test_trainer_reset_kv didn't fail for 10K times. 3rd Trigger. * remove for loop
apache · Jun 25, 2018 · 619e4bd · 619e4bd
1 parent cdb01fc
commit 619e4bd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/python/unittest/test_gluon_trainer.py b/tests/python/unittest/test_gluon_trainer.py
@@ -190,6 +190,7 @@ def check_trainer_reset_kv(kv):
         trainer.step(1)
         assert trainer._kvstore.type == kv
         # load would reset kvstore
+        mx.nd.waitall()
         params.load('test_trainer_reset_kv.params')
         assert trainer._kvstore is None
         assert trainer._kv_initialized is False