Skip to content

GitLab

Commit 3e12ad2b authored by Yanhui Liang's avatar Yanhui Liang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 292230496
parent d948ff3f
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
official/benchmark/bert_squad_benchmark.py
View file @ 3e12ad2b
...@@ -481,7 +481,6 @@ class BertSquadMultiWorkerAccuracy(BertSquadBenchmarkBase): ...@@ -481,7 +481,6 @@ class BertSquadMultiWorkerAccuracy(BertSquadBenchmarkBase):
num_gpus = 8 num_gpus = 8
FLAGS.num_gpus = num_gpus FLAGS.num_gpus = num_gpus
FLAGS.dtype = 'fp16' FLAGS.dtype = 'fp16'
FLAGS.enable_eager = True,
FLAGS.enable_xla = False FLAGS.enable_xla = False
FLAGS.distribution_strategy = 'multi_worker_mirrored' FLAGS.distribution_strategy = 'multi_worker_mirrored'
FLAGS.tf_gpu_thread_mode = 'gpu_private' FLAGS.tf_gpu_thread_mode = 'gpu_private'
...@@ -554,7 +553,6 @@ class BertSquadMultiWorkerBenchmark(BertSquadBenchmarkBase): ...@@ -554,7 +553,6 @@ class BertSquadMultiWorkerBenchmark(BertSquadBenchmarkBase):
num_gpus = 8 num_gpus = 8
FLAGS.num_gpus = num_gpus FLAGS.num_gpus = num_gpus
FLAGS.dtype = 'fp16' FLAGS.dtype = 'fp16'
FLAGS.enable_eager = True
FLAGS.enable_xla = False FLAGS.enable_xla = False
FLAGS.distribution_strategy = 'multi_worker_mirrored' FLAGS.distribution_strategy = 'multi_worker_mirrored'
FLAGS.tf_gpu_thread_mode = 'gpu_private' FLAGS.tf_gpu_thread_mode = 'gpu_private'
......
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