pbar correctly display the CTC decoding progress

haotianteng · Aug 24, 2018 · 7da4d4e · 7da4d4e
1 parent a444e8f
commit 7da4d4e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)