From 1ac04372b92a47c9191fc984a86382562f549ce0 Mon Sep 17 00:00:00 2001
From: Dirk Groeneveld <dirkg@allenai.org>
Date: Mon, 4 Nov 2024 13:18:19 -0800
Subject: [PATCH] Bounce epoch

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

diff --git a/scripts/augusta/peteish13-highlr-anneal-launch.sh b/scripts/augusta/peteish13-highlr-anneal-launch.sh
index b2214174f..40d04204c 100755
--- a/scripts/augusta/peteish13-highlr-anneal-launch.sh
+++ b/scripts/augusta/peteish13-highlr-anneal-launch.sh
@@ -39,6 +39,7 @@ mkdir -p $SAVE_FOLDER
     --compile.fullgraph=false \
     --fused_loss=true \
     --model.flash_attention=true \
+    --epoch=1 \
     --data.num_workers=8 \
     --optimizer.metrics_log_interval=10 \
     --data.prefetch_factor=8 2>&1 | tee $SAVE_FOLDER/log.txt