Skip to content

GitLab

"git@developer.sourcefind.cn:ox696c/ktransformers.git" did not exist on "c92a300bd8bee05144add05365e24b6cb2db82e7"
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): ...@@ -125,8 +125,8 @@ class BertPretrainAccuracyBenchmark(bert_benchmark_utils.BertBenchmarkBase):
# This is used for accuracy test. # This is used for accuracy test.
self._setup() self._setup()
self._specify_common_flags() self._specify_common_flags()
FLAGS.num_steps_per_epoch = 25000 FLAGS.num_steps_per_epoch = 250000
FLAGS.num_train_epochs = 40 FLAGS.num_train_epochs = 4
FLAGS.model_dir = self._get_model_dir( FLAGS.model_dir = self._get_model_dir(
'benchmark_8x8_tpu_bf16_seq128_1m_steps') 'benchmark_8x8_tpu_bf16_seq128_1m_steps')
summary_path = os.path.join(FLAGS.model_dir, 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