提交 62a87b39 编写于 作者: C chenjiawen

修改README,添加数据示例

上级 c230371b
此差异已折叠。
此差异已折叠。
此差异已折叠。
export CUDA_VISIBLE_DEVICES=4 export CUDA_VISIBLE_DEVICES=0
python train.py \ python train.py \
--train_path='data/train/sentence_file_*' \ --train_path='data/train/sentence_file_*' \
--test_path='data/dev/sentence_file_*' \ --test_path='data/dev/sentence_file_*' \
......
...@@ -578,8 +578,6 @@ def train_loop(args, ...@@ -578,8 +578,6 @@ def train_loop(args,
end_time = time.time() end_time = time.time()
total_time += end_time - start_time total_time += end_time - start_time
logger.info("train ppl {}".format(ppl))
if epoch_id == args.max_epoch - 1 and args.enable_ce: if epoch_id == args.max_epoch - 1 and args.enable_ce:
logger.info("lstm_language_model_duration\t%s" % logger.info("lstm_language_model_duration\t%s" %
(total_time / args.max_epoch)) (total_time / args.max_epoch))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册