Update train.py

2025-04-01 11:06:12 +02:00 · 2025-04-01 11:06:12 +02:00 · 467671ffca
commit 467671ffca
parent e3e5deb32f
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -41,6 +41,7 @@ training_args = TrainingArguments(
    warmup_steps=500,
    weight_decay=0.01,
    logging_dir='./logs',
+    save_total_limit=5,
    fp16=True  # Enable Mixed Precision Training for GPU
 )