Internal change

PiperOrigin-RevId: 438075627

Internal change
PiperOrigin-RevId: 438075627
368f3305 · A. Unique TensorFlower · b3f57016 · 368f3305
Commit 368f3305 authored Mar 29, 2022 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

official/benchmark/resnet50_keras_core.py official/benchmark/resnet50_keras_core.py +8 -0

No files found.
--- a/official/benchmark/resnet50_keras_core.py
+++ b/official/benchmark/resnet50_keras_core.py
@@ -111,6 +111,14 @@ class Resnet50KerasCoreBenchmark(perfzero_benchmark.PerfZeroBenchmark):
    max_wall_time = max(wall_times)
    self.report_benchmark(iters=-1, wall_time=max_wall_time)

+  def benchmark_1_gpu_min_3(self):
+    num_trials = 3
+    wall_times = []
+    for _ in range(num_trials):
+      wall_times.append(_run_benchmark())
+    min_wall_time = min(wall_times)
+    self.report_benchmark(iters=-1, wall_time=min_wall_time)
+
  def benchmark_1_gpu_med_3(self):
    num_trials = 3
    wall_times = []