fix(eval): steps

AdityaNG · Mar 3, 2024 · 7979436 · 7979436
1 parent 7a4ba8e
commit 7979436
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/drivellava/scripts/train.py b/drivellava/scripts/train.py
@@ -150,7 +150,7 @@ def main():
         --per_device_eval_batch_size 4 \
         --gradient_accumulation_steps 1 \
         --evaluation_strategy "steps" \
-        --eval_steps 15 \
+        --eval_steps 50 \
         --save_strategy "steps" \
         --save_steps 50 \
         --save_total_limit 1 \