Internal change

PiperOrigin-RevId: 282857541

Internal change
PiperOrigin-RevId: 282857541
0effd158 · Sai Ganesh Bandiatmakuri · A. Unique TensorFlower · 61d6173d · 0effd158
Commit 0effd158 authored Nov 27, 2019 by Sai Ganesh Bandiatmakuri Committed by A. Unique TensorFlower Nov 27, 2019
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

official/staging/shakespeare/shakespeare_benchmark.py official/staging/shakespeare/shakespeare_benchmark.py +4 -3

No files found.
--- a/official/staging/shakespeare/shakespeare_benchmark.py
+++ b/official/staging/shakespeare/shakespeare_benchmark.py
@@ -88,6 +88,7 @@ class ShakespeareBenchmarkBase(PerfZeroBenchmark):
      elapsed = time_log[-1].timestamp - time_log[warmup].timestamp
      num_examples = (
          total_batch_size * log_steps * (len(time_log) - warmup - 1))
+      if elapsed > 0:
        examples_per_sec = num_examples / elapsed
        metrics.append({'name': 'exp_per_second',
                        'value': examples_per_sec})