From 8b7afd0e7d1b4cff21eab34193fe5930b4a5bb18 Mon Sep 17 00:00:00 2001 From: epwalsh Date: Thu, 25 Jul 2024 11:41:17 -0700 Subject: [PATCH] final tweaks --- configs/peteish1-weka.yaml | 2 +- configs/peteish7-weka.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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