Internal change

PiperOrigin-RevId: 393376715

Internal change
PiperOrigin-RevId: 393376715
14f82a3d · A. Unique TensorFlower · f8d69319 · 14f82a3d
Commit 14f82a3d authored Aug 27, 2021 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/benchmark/retinanet_benchmark.py official/benchmark/retinanet_benchmark.py +1 -1

No files found.
--- a/official/benchmark/retinanet_benchmark.py
+++ b/official/benchmark/retinanet_benchmark.py
@@ -287,7 +287,7 @@ class DetectionBenchmarkReal(DetectionAccuracy):
    params['train']['iterations_per_loop'] = 500
    FLAGS.model_dir = self._get_model_dir('real_benchmark_4x4_tpu_coco')
    FLAGS.strategy_type = 'tpu'
-    self._run_and_report_benchmark(params, do_eval=False, warmup=0)
+    self._run_and_report_benchmark(params, do_eval=False, warmup=1)

  @flagsaver.flagsaver
  def benchmark_2x2_tpu_coco_mlir(self):