Unverified Commit ac47458a authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[s2s] round runtime in run_eval (#6798)

parent 5ab21b07
......@@ -67,7 +67,7 @@ def generate_summaries_or_translations(
fout.write(hypothesis + "\n")
fout.flush()
fout.close()
runtime = time.time() - start_time
runtime = int(time.time() - start_time) # seconds
n_obs = len(examples)
return dict(n_obs=n_obs, runtime=runtime, seconds_per_sample=round(runtime / n_obs, 4))
......
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