Skip to content

GitLab

Commit 0effd158 authored by Sai Ganesh Bandiatmakuri's avatar Sai Ganesh Bandiatmakuri Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 282857541
parent 61d6173d
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
official/staging/shakespeare/shakespeare_benchmark.py
View file @ 0effd158
...@@ -88,9 +88,10 @@ class ShakespeareBenchmarkBase(PerfZeroBenchmark): ...@@ -88,9 +88,10 @@ class ShakespeareBenchmarkBase(PerfZeroBenchmark):
elapsed = time_log[-1].timestamp - time_log[warmup].timestamp elapsed = time_log[-1].timestamp - time_log[warmup].timestamp
num_examples = ( num_examples = (
total_batch_size * log_steps * (len(time_log) - warmup - 1)) total_batch_size * log_steps * (len(time_log) - warmup - 1))
examples_per_sec = num_examples / elapsed if elapsed > 0:
metrics.append({'name': 'exp_per_second', examples_per_sec = num_examples / elapsed
'value': examples_per_sec}) metrics.append({'name': 'exp_per_second',
'value': examples_per_sec})
flags_str = flags_core.get_nondefault_flags_as_str() flags_str = flags_core.get_nondefault_flags_as_str()
self.report_benchmark(iters=-1, wall_time=wall_time_sec, self.report_benchmark(iters=-1, wall_time=wall_time_sec,
......
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