Skip to content

GitLab

Commit f4dd2cff authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 285223338
parent 152adcbd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/benchmark/retinanet_benchmark.py
View file @ f4dd2cff
...@@ -260,7 +260,7 @@ class RetinanetBenchmarkReal(RetinanetAccuracy): ...@@ -260,7 +260,7 @@ class RetinanetBenchmarkReal(RetinanetAccuracy):
FLAGS.num_gpus = self.num_gpus FLAGS.num_gpus = self.num_gpus
FLAGS.params_override = json.dumps(params) FLAGS.params_override = json.dumps(params)
FLAGS.model_dir = self._get_model_dir('real_benchmark_1_gpu_coco') FLAGS.model_dir = self._get_model_dir('real_benchmark_1_gpu_coco')
FLAGS.strategy_type = 'one_device_gpu' FLAGS.strategy_type = 'one_device'
# Use negative value to avoid saving checkpoints. # Use negative value to avoid saving checkpoints.
FLAGS.save_checkpoint_freq = -1 FLAGS.save_checkpoint_freq = -1
if self.timer_callback is None: if self.timer_callback is None:
...@@ -281,7 +281,7 @@ class RetinanetBenchmarkReal(RetinanetAccuracy): ...@@ -281,7 +281,7 @@ class RetinanetBenchmarkReal(RetinanetAccuracy):
FLAGS.num_gpus = self.num_gpus FLAGS.num_gpus = self.num_gpus
FLAGS.params_override = json.dumps(params) FLAGS.params_override = json.dumps(params)
FLAGS.model_dir = self._get_model_dir('real_benchmark_1_gpu_coco') FLAGS.model_dir = self._get_model_dir('real_benchmark_1_gpu_coco')
FLAGS.strategy_type = 'one_device_gpu' FLAGS.strategy_type = 'one_device'
FLAGS.enable_xla = True FLAGS.enable_xla = True
# Use negative value to avoid saving checkpoints. # Use negative value to avoid saving checkpoints.
FLAGS.save_checkpoint_freq = -1 FLAGS.save_checkpoint_freq = -1
......
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