cluster-rnn

commit 950feb3caa142ba17f3641265d39cde068b8388f
parent db8271c1cf1b0dbbd1d6b11198b22bde8c3648d8
Author: umhau <umhau@users.noreply.github.com>
Date:   Wed, 15 Feb 2017 16:02:34 -0500

sddsffd

Diffstat:
M
train.lua
 | 
14
+++++++-------

1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/train.lua b/train.lua
@@ -412,14 +412,14 @@ for i = 1, iterations do
         local savefile = string.format('%s/lm_%s_epoch%.2f_%.4f.t7', opt.checkpoint_dir, opt.savefile, epoch, val_loss)
         print('saving checkpoint to ' .. savefile)
         local checkpoint = {}
-        checkpoint.protos = protos
+        --checkpoint.protos = protos
         --checkpoint.opt = opt
-        checkpoint.train_losses = train_losses
-        checkpoint.val_loss = val_loss
-        checkpoint.val_losses = val_losses
-        checkpoint.i = i
-        checkpoint.epoch = epoch
-        checkpoint.vocab = loader.vocab_mapping
+        --checkpoint.train_losses = train_losses
+        --checkpoint.val_loss = val_loss
+        --checkpoint.val_losses = val_losses
+        --checkpoint.i = i
+        --checkpoint.epoch = epoch
+        --checkpoint.vocab = loader.vocab_mapping
         --checkpoint.optim_state = optim_state
         --checkpoint.optimizer = opt.optimizer
         torch.save(savefile, checkpoint)