提交 0cc61be6 编写于 作者: Z zhangxuefei

Add the difination epoch_id before its reference

上级 1df04ca0
...@@ -555,6 +555,7 @@ def train_loop(args, ...@@ -555,6 +555,7 @@ def train_loop(args,
valid_ppl = eval(vocab, infer_progs, dev_count, logger, args) valid_ppl = eval(vocab, infer_progs, dev_count, logger, args)
logger.info("valid ppl {}".format(valid_ppl)) logger.info("valid ppl {}".format(valid_ppl))
if batch_id > 0 and batch_id % args.save_interval == 0: 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, model_path = os.path.join(args.para_save_dir,
str(batch_id + epoch_id)) str(batch_id + epoch_id))
if not os.path.isdir(model_path): if not os.path.isdir(model_path):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册