Skip to content

GitLab

Commit 297a4bbc authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 377390629
parent 88a6cf61
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/benchmark/retinanet_benchmark.py
View file @ 297a4bbc
...@@ -279,11 +279,11 @@ class DetectionBenchmarkReal(DetectionAccuracy): ...@@ -279,11 +279,11 @@ class DetectionBenchmarkReal(DetectionAccuracy):
@flagsaver.flagsaver @flagsaver.flagsaver
def benchmark_4x4_tpu_coco(self): def benchmark_4x4_tpu_coco(self):
"""Run detection model accuracy test with 4 TPUs.""" """Run detection model accuracy test with 4x4 TPU."""
self._setup() self._setup()
params = self._params() params = self._params()
params['train']['batch_size'] = 256 params['train']['batch_size'] = 256
params['train']['total_steps'] = 469 # One epoch. params['train']['total_steps'] = 10 * 469 # 10 epochs.
params['train']['iterations_per_loop'] = 500 params['train']['iterations_per_loop'] = 500
FLAGS.model_dir = self._get_model_dir('real_benchmark_4x4_tpu_coco') FLAGS.model_dir = self._get_model_dir('real_benchmark_4x4_tpu_coco')
FLAGS.strategy_type = 'tpu' FLAGS.strategy_type = 'tpu'
......
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