cluster-rnn

commit 589bba0e4e37c9b83cd7242b377cd5955f0f8114
parent b5047961995ec865134708e5b29f3ecb2d241a80
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 14:19:05 -0500

ditto

Diffstat:
M
train.lua
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.lua b/train.lua
@@ -420,7 +420,7 @@ for i = 1, iterations do
         checkpoint.i = i
         checkpoint.epoch = epoch
         checkpoint.vocab = loader.vocab_mapping
-        checkpoint.optim_state = optim_state
+        --checkpoint.optim_state = optim_state
         checkpoint.optimizer = opt.optimizer
         torch.save(savefile, checkpoint)
     end