From 0cc61be6edb99e0c4b2d58608cfca4c16e15ae0b Mon Sep 17 00:00:00 2001
From: zhangxuefei <zhangxuefei@baidu.com>
Date: Mon, 6 May 2019 16:58:59 +0800
Subject: [PATCH] Add the difination epoch_id before its reference

---
 ELMo/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ELMo/train.py b/ELMo/train.py
index c8726f8..c062df3 100755
--- a/ELMo/train.py
+++ b/ELMo/train.py
@@ -555,6 +555,7 @@ def train_loop(args,
             valid_ppl = eval(vocab, infer_progs, dev_count, logger, args)
             logger.info("valid ppl {}".format(valid_ppl))
         if batch_id > 0 and batch_id % args.save_interval == 0:
+            epoch_id = int(batch_id / n_batches_per_epoch)
             model_path = os.path.join(args.para_save_dir,
                                       str(batch_id + epoch_id))
             if not os.path.isdir(model_path):
-- 
GitLab