Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "19d4eaaf5e1883776b341a4f704bb6a277e315aa"
Commit 3928d481 authored by Igor's avatar Igor Committed by guptapriya
Browse files

Make 'off' a string literal.

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