diff --git a/train.py b/train.py
index 1754e4e..5d2d76e 100644
--- 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
 )