From 7da4d4e6a63891749e7a46beb39f318d153ec65f Mon Sep 17 00:00:00 2001 From: Haotian Teng Date: Sat, 25 Aug 2018 07:32:14 +1000 Subject: [PATCH] pbar correctly display the CTC decoding progress --- chiron/chiron_eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/chiron/chiron_eval.py b/chiron/chiron_eval.py index a08663d..15beb6f 100755 --- a/chiron/chiron_eval.py +++ b/chiron/chiron_eval.py @@ -298,9 +298,9 @@ def worker_fn(): decoded_fname, i, predict_val, logits_prob = sess.run(decode_ops, feed_dict={training: False}) decoded_fname = decoded_fname.decode("UTF-8") val[decoded_fname][i] = (predict_val, logits_prob) + pbars.update(1,progress = len(val[name])*FLAGS.batch_size) + pbars.update_bar() if len(val[name]) == N: - pbars.update(1,progress = len(val[name])*FLAGS.batch_size) - pbars.update_bar() break pbars.update(3,progress = f_i+1)