Skip to content

M
Models

曾经的那一瞬间 / Models
11 个月 前同步成功

通知 1
Star 0
Fork 0
M
Models

体验新版 GitCode，发现更多精彩内容 >>

提交 f4dd2cff 编写于 作者: Y Yeqing Li 提交者： A. Unique TensorFlower
浏览文件
操作

Internal change 

PiperOrigin-RevId: 285223338
上级 152adcbd
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
official/benchmark/retinanet_benchmark.py
浏览文件 @ f4dd2cff
......@@ -260,7 +260,7 @@ class RetinanetBenchmarkReal(RetinanetAccuracy):
FLAGS.num_gpus = self.num_gpus
FLAGS.params_override = json.dumps(params)
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.
FLAGS.save_checkpoint_freq = -1
if self.timer_callback is None:
......@@ -281,7 +281,7 @@ class RetinanetBenchmarkReal(RetinanetAccuracy):
FLAGS.num_gpus = self.num_gpus
FLAGS.params_override = json.dumps(params)
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
# Use negative value to avoid saving checkpoints.
FLAGS.save_checkpoint_freq = -1
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或