Skip to content

GitLab

Commit e80b385a authored by guptapriya's avatar guptapriya Committed by guptapriya
Browse files

update estimator benchmarks too

parent 39638d66
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/transformer/transformer_estimator_benchmark.py
View file @ e80b385a
......@@ -174,7 +174,7 @@ class TransformerBigEstimatorAccuracy(EstimatorBenchmark):
FLAGS.param_set = 'big'
FLAGS.batch_size = 3072 * 8
FLAGS.static_batch = True
FLAGS.max_length = 40
FLAGS.max_length = 64
FLAGS.train_steps = 100000
FLAGS.steps_between_evals = 5000
FLAGS.model_dir = self._get_model_dir('benchmark_graph_8_gpu')
......@@ -294,7 +294,7 @@ class TransformerBaseEstimatorAccuracy(EstimatorBenchmark):
FLAGS.param_set = 'base'
FLAGS.batch_size = 4096 * 8
FLAGS.static_batch = True
FLAGS.max_length = 40
FLAGS.max_length = 64
FLAGS.train_steps = 100000
FLAGS.steps_between_evals = 5000
FLAGS.model_dir = self._get_model_dir('benchmark_graph_8_gpu')
......
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