Update bert.yaml to be consistent with mlperf storage

argonne-lcf · Jul 27, 2023 · 25ba2a0 · 25ba2a0
1 parent b7b2b92
commit 25ba2a0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dlio_benchmark/configs/workload/bert.yaml b/dlio_benchmark/configs/workload/bert.yaml
@@ -17,8 +17,9 @@ dataset:
   file_prefix: part
 
 train:
+  seed_change_epoch: False
   computation_time: 0.968
-  total_training_steps: 5000
+  total_training_steps: 1000
 
 reader:
   data_loader: tensorflow
@@ -31,5 +32,5 @@ reader:
 
 checkpoint:
   checkpoint_folder: checkpoints/bert
-  steps_between_checkpoints: 1250
+  steps_between_checkpoints: 250
   model_size: 4034713312