final tweaks

allenai · Jul 25, 2024 · 8b7afd0 · 8b7afd0
1 parent accc3ee
commit 8b7afd0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/peteish1-weka.yaml b/configs/peteish1-weka.yaml
@@ -83,7 +83,7 @@ save_num_unsharded_checkpoints_to_keep: -1
 
 load_path: null
 
-max_duration: 2ep
+max_duration: 1ep
 global_train_batch_size: 1024
 device_train_microbatch_size: 4
 

diff --git a/configs/peteish7-weka.yaml b/configs/peteish7-weka.yaml
@@ -83,7 +83,7 @@ save_num_unsharded_checkpoints_to_keep: -1
 
 load_path: null
 
-max_duration: 2ep
+max_duration: 1ep
 global_train_batch_size: 1024
 device_train_microbatch_size: 2