This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 619e4bd Fix #11353 (#11360)
619e4bd is described below
commit 619e4bded058e4bf77029bc30b395275d72f7907
Author: Haibin Lin <[email protected]>
AuthorDate: Mon Jun 25 15:34:54 2018 -0700
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
---
tests/python/unittest/test_gluon_trainer.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/tests/python/unittest/test_gluon_trainer.py
b/tests/python/unittest/test_gluon_trainer.py
index 1c59cea..eac9fad 100644
--- a/tests/python/unittest/test_gluon_trainer.py
+++ b/tests/python/unittest/test_gluon_trainer.py
@@ -190,6 +190,7 @@ def test_trainer_reset_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
