Bounce epoch

allenai · Nov 4, 2024 · 1ac0437 · 1ac0437
1 parent bdb6c13
commit 1ac0437
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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