Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "3e6c2610ae9256dc7e4118dbf2074e97487babe3"
Commit 99a34a6c authored by guptapriya's avatar guptapriya
Browse files

fix lint

parent 97c1e898
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/transformer/v2/transformer_benchmark.py
View file @ 99a34a6c
...@@ -328,8 +328,7 @@ class TransformerKerasBenchmark(TransformerBenchmark): ...@@ -328,8 +328,7 @@ class TransformerKerasBenchmark(TransformerBenchmark):
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( FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu_no_ds_sb')
'benchmark_1_gpu_no_dist_strat_static_batch')
FLAGS.static_batch = True FLAGS.static_batch = True
FLAGS.max_length = 64 FLAGS.max_length = 64
self._run_and_report_benchmark(total_batch_size=FLAGS.batch_size, self._run_and_report_benchmark(total_batch_size=FLAGS.batch_size,
......
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