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
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
official/staging/shakespeare/shakespeare_benchmark.py
View file @ 0effd158
...@@ -88,6 +88,7 @@ class ShakespeareBenchmarkBase(PerfZeroBenchmark): ...@@ -88,6 +88,7 @@ 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))
if elapsed > 0:
examples_per_sec = num_examples / elapsed examples_per_sec = num_examples / elapsed
metrics.append({'name': 'exp_per_second', metrics.append({'name': 'exp_per_second',
'value': examples_per_sec}) 'value': examples_per_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