change args.checkpoint_steps from 1000 to 1 for test evaluate func in…

… VAE (#137)
thu-coai · Jul 1, 2019 · 9f87ea4 · 9f87ea4
1 parent 3504ab2
commit 9f87ea4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/models/VAE-tensorflow/test_VAE_tensorflow.py b/models/VAE-tensorflow/test_VAE_tensorflow.py
@@ -56,6 +56,7 @@ def modify_args(args):
 	args.name = 'test_VAE_tensorflow'
 	args.wvclass = 'Glove'
 	args.epochs = 5
+	args.checkpoint_steps = 1
 	args.datapath = path + '/tests/dataloader/dummy_mscoco'
 
 def test_train(mocker):