Skip to content

GitLab

Commit bf47b956 authored by Alexei Baevski's avatar Alexei Baevski Committed by Myle Ott
Browse files

also report sentence/s timing when generating

parent 67af40c9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
generate.py
View file @ bf47b956
......@@ -157,8 +157,8 @@ def main(args):
t.log({'wps': round(wps_meter.avg)})
num_sentences += 1
print('| Translated {} sentences ({} tokens) in {:.1f}s ({:.2f} tokens/s)'.format(
num_sentences, gen_timer.n, gen_timer.sum, 1. / gen_timer.avg))
print('| Translated {} sentences ({} tokens) in {:.1f}s ({:.2f} sentences/s, {:.2f} tokens/s)'.format(
num_sentences, gen_timer.n, gen_timer.sum, num_sentences/ gen_timer.sum, 1. / gen_timer.avg))
if has_target:
print('| Generate {} with beam={}: {}'.format(args.gen_subset, args.beam, scorer.result_string()))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment