Skip to content

GitLab

Commit 89f4a1a6 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 309095156
parent ec7fbf0d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/benchmark/bert_pretrain_benchmark.py
View file @ 89f4a1a6
......@@ -125,8 +125,8 @@ class BertPretrainAccuracyBenchmark(bert_benchmark_utils.BertBenchmarkBase):
# This is used for accuracy test.
self._setup()
self._specify_common_flags()
FLAGS.num_steps_per_epoch = 25000
FLAGS.num_train_epochs = 40
FLAGS.num_steps_per_epoch = 250000
FLAGS.num_train_epochs = 4
FLAGS.model_dir = self._get_model_dir(
'benchmark_8x8_tpu_bf16_seq128_1m_steps')
summary_path = os.path.join(FLAGS.model_dir,
......
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