提交 4c148e61 编写于 作者: X xuezhong

change run.sh

上级 69199db5
...@@ -385,8 +385,9 @@ def evaluate(logger, args): ...@@ -385,8 +385,9 @@ def evaluate(logger, args):
] ]
feeder = fluid.DataFeeder(feed_list, place) feeder = fluid.DataFeeder(feed_list, place)
inference_program = main_program.clone(for_test=True)
eval_loss, bleu_rouge = validation( eval_loss, bleu_rouge = validation(
main_program, avg_cost, s_probs, e_probs, inference_program, avg_cost, s_probs, e_probs,
feed_order, place, vocab, brc_data, logger, args) feed_order, place, vocab, brc_data, logger, args)
logger.info('Dev eval loss {}'.format(eval_loss)) logger.info('Dev eval loss {}'.format(eval_loss))
logger.info('Dev eval result: {}'.format(bleu_rouge)) logger.info('Dev eval result: {}'.format(bleu_rouge))
...@@ -431,8 +432,9 @@ def predict(logger, args): ...@@ -431,8 +432,9 @@ def predict(logger, args):
] ]
feeder = fluid.DataFeeder(feed_list, place) feeder = fluid.DataFeeder(feed_list, place)
inference_program = main_program.clone(for_test=True)
eval_loss, bleu_rouge = validation( eval_loss, bleu_rouge = validation(
main_program, avg_cost, s_probs, e_probs, inference_program, avg_cost, s_probs, e_probs,
feed_order, place, vocab, brc_data, logger, args) feed_order, place, vocab, brc_data, logger, args)
......
...@@ -17,6 +17,4 @@ python run.py \ ...@@ -17,6 +17,4 @@ python run.py \
--max_p_len 500 \ --max_p_len 500 \
--max_q_len 60 \ --max_q_len 60 \
--max_a_len 200 \ --max_a_len 200 \
--drop_rate 0.2 \ --drop_rate 0.2 $@\
--train \
--test
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册