From 949d80c639a6ce381135afce1644fadf9c62ae61 Mon Sep 17 00:00:00 2001
From: Dirk Groeneveld <dirkg@allenai.org>
Date: Thu, 14 Nov 2024 21:19:29 -0800
Subject: [PATCH] Don't run out of space.

---
 scripts/augusta/beaker/peteish13.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scripts/augusta/beaker/peteish13.sh b/scripts/augusta/beaker/peteish13.sh
index 3f5b09604..2aad8fde6 100755
--- a/scripts/augusta/beaker/peteish13.sh
+++ b/scripts/augusta/beaker/peteish13.sh
@@ -68,6 +68,7 @@ torchrun \
       --run_name=$RUN_NAME \
       --wandb.group=$NAME \
       --save_interval_ephemeral=1000 \
+      --save_num_checkpoints_to_keep=2 \
       --eval_interval=1000 \
       --fsdp.sharding_strategy=HYBRID_SHARD \
       --fsdp.hybrid_sharding_num_model_replicas="${BEAKER_REPLICA_COUNT}" \