Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "7812a026f7c6605e954af9b7fa248da606513846"
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): ...@@ -174,7 +174,7 @@ class TransformerBigEstimatorAccuracy(EstimatorBenchmark):
FLAGS.param_set = 'big' FLAGS.param_set = 'big'
FLAGS.batch_size = 3072 * 8 FLAGS.batch_size = 3072 * 8
FLAGS.static_batch = True FLAGS.static_batch = True
FLAGS.max_length = 40 FLAGS.max_length = 64
FLAGS.train_steps = 100000 FLAGS.train_steps = 100000
FLAGS.steps_between_evals = 5000 FLAGS.steps_between_evals = 5000
FLAGS.model_dir = self._get_model_dir('benchmark_graph_8_gpu') FLAGS.model_dir = self._get_model_dir('benchmark_graph_8_gpu')
...@@ -294,7 +294,7 @@ class TransformerBaseEstimatorAccuracy(EstimatorBenchmark): ...@@ -294,7 +294,7 @@ class TransformerBaseEstimatorAccuracy(EstimatorBenchmark):
FLAGS.param_set = 'base' FLAGS.param_set = 'base'
FLAGS.batch_size = 4096 * 8 FLAGS.batch_size = 4096 * 8
FLAGS.static_batch = True FLAGS.static_batch = True
FLAGS.max_length = 40 FLAGS.max_length = 64
FLAGS.train_steps = 100000 FLAGS.train_steps = 100000
FLAGS.steps_between_evals = 5000 FLAGS.steps_between_evals = 5000
FLAGS.model_dir = self._get_model_dir('benchmark_graph_8_gpu') 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