Skip to content

GitLab

Commit 2be9ba5b authored by guptapriya's avatar guptapriya
Browse files

Turn dist strat off for 1 GPU benchmarks

parent 1d16f473
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/transformer/v2/transformer_benchmark.py
View file @ 2be9ba5b
......@@ -133,6 +133,7 @@ class TransformerBaseKerasAccuracy(TransformerBenchmark):
"""
self._setup()
FLAGS.num_gpus = 1
FLAGS.distribution_strategy = off
FLAGS.data_dir = self.train_data_dir
FLAGS.vocab_file = self.vocab_file
# Sets values directly to avoid validation check.
......@@ -158,6 +159,7 @@ class TransformerBaseKerasAccuracy(TransformerBenchmark):
"""
self._setup()
FLAGS.num_gpus = 1
FLAGS.distribution_strategy = off
FLAGS.data_dir = self.train_data_dir
FLAGS.vocab_file = self.vocab_file
# Sets values directly to avoid validation check.
......@@ -315,6 +317,7 @@ class TransformerKerasBenchmark(TransformerBenchmark):
"""Benchmark 1 gpu."""
self._setup()
FLAGS.num_gpus = 1
FLAGS.distribution_strategy = off
FLAGS.batch_size = self.batch_per_gpu
FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu')
self._run_and_report_benchmark(total_batch_size=FLAGS.batch_size,
......@@ -324,6 +327,7 @@ class TransformerKerasBenchmark(TransformerBenchmark):
"""Benchmark 1 gpu."""
self._setup()
FLAGS.num_gpus = 1
FLAGS.distribution_strategy = off
FLAGS.batch_size = self.batch_per_gpu
FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu_static_batch')
# TODO(guptapriya): Add max_length
......
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