提交 6ba4732b 编写于 作者: X xuezhong

fix bug for test

上级 a8cd574b
...@@ -94,7 +94,7 @@ def parse_args(): ...@@ -94,7 +94,7 @@ def parse_args():
parser.add_argument('--cell_clip', type=float, default=3.0) parser.add_argument('--cell_clip', type=float, default=3.0)
parser.add_argument('--max_epoch', type=float, default=10) parser.add_argument('--max_epoch', type=float, default=10)
parser.add_argument('--local', type=bool, default=False) parser.add_argument('--local', type=bool, default=False)
parser.add_argument('--shuffle', type=bool, default=False) parser.add_argument('--shuffle', type=bool, default=True)
parser.add_argument('--use_custom_samples', type=bool, default=False) parser.add_argument('--use_custom_samples', type=bool, default=False)
parser.add_argument('--para_save_dir', type=str, default='model_new') parser.add_argument('--para_save_dir', type=str, default='model_new')
parser.add_argument('--train_path', type=str, default='') parser.add_argument('--train_path', type=str, default='')
......
...@@ -5,4 +5,5 @@ python train.py \ ...@@ -5,4 +5,5 @@ python train.py \
--vocab_path baike/vocabulary_min5k.txt \ --vocab_path baike/vocabulary_min5k.txt \
--learning_rate 0.2 \ --learning_rate 0.2 \
--use_gpu True \ --use_gpu True \
--shuffle True \
--local True $@ --local True $@
...@@ -578,7 +578,6 @@ def train_loop(args, ...@@ -578,7 +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" %
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册