Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Fix #11353 #11360

Merged
merged 12 commits into from
Jun 25, 2018
1 change: 1 addition & 0 deletions tests/python/unittest/test_gluon_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down