From 357aab2555ed41e805a3f8cb2dc4cc148a13a7b8 Mon Sep 17 00:00:00 2001
From: mdasadul <mdasadul@ualberta.ca>
Date: Tue, 27 Mar 2018 08:50:55 -0400
Subject: [PATCH] saving checkpoint after fininshing an epoch and before
 decode/external evaluation

---
 nmt/train.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/nmt/train.py b/nmt/train.py
index 75978ec4..fbfe4c3b 100644
--- a/nmt/train.py
+++ b/nmt/train.py
@@ -357,6 +357,13 @@ def train(hparams, scope=None, target_session=""):
       utils.print_out(
           "# Finished an epoch, step %d. Perform external evaluation" %
           global_step)
+      
+      # Save checkpoint
+      loaded_train_model.saver.save(
+          train_sess,
+          os.path.join(out_dir, "translate.ckpt"),
+          global_step=global_step)
+
       run_sample_decode(infer_model, infer_sess, model_dir, hparams,
                         summary_writer, sample_src_data, sample_tgt_data)
       run_external_eval(infer_model, infer_sess, model_dir, hparams,