Skip to content

GitLab

"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "9dcdf5da0333b3592cccf0285ad092f9eaf5abfe"
Commit 194edbf3 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 311155771
parent ca0ccba7
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/benchmark/bert_pretrain_benchmark.py
View file @ 194edbf3
...@@ -131,8 +131,8 @@ class BertPretrainAccuracyBenchmark(bert_benchmark_utils.BertBenchmarkBase): ...@@ -131,8 +131,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 = 250000 FLAGS.num_steps_per_epoch = 1000000
FLAGS.num_train_epochs = 4 FLAGS.num_train_epochs = 1
FLAGS.model_dir = self._get_model_dir( FLAGS.model_dir = self._get_model_dir(
'benchmark_accuracy_8x8_tpu_bf16_seq128_1m_steps') 'benchmark_accuracy_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