diff --git a/configs/peteish1-weka.yaml b/configs/peteish1-weka.yaml index 48452318d..071c5399b 100644 --- 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 index 5137fb439..a7dc9d66c 100644 --- 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