cluster-rnn

commit cdd2bfe1219d8f613de96cfd049b38b03602da13
parent c41d01c9ec096cec3d753b8d106645ce9147473e
Author: umhau <umhau@users.noreply.github.com>
Date:   Thu, 16 Feb 2017 13:14:28 -0500

enable opt field in checkpoint

Diffstat:
M
train.lua
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.lua b/train.lua
@@ -415,7 +415,7 @@ for i = 1, iterations do
         print('saving checkpoint to ' .. savefile)
         local checkpoint = {}
         checkpoint.protos = protos
-        --checkpoint.opt = opt
+        checkpoint.opt = opt
         checkpoint.train_losses = train_losses
         checkpoint.val_loss = val_loss
         checkpoint.val_losses = val_losses