Update train.py

train.py

@@ -326,9 +326,9 @@ def main():
         weight_decay=0.01,
         lr_scheduler_type="cosine",
         warmup_ratio=0.1,
-        load_best_model_at_end=True,
-        metric_for_best_model="eval_loss",  # Use standard eval_loss
-        greater_is_better=False,  # Lower loss is better
+        load_best_model_at_end=False,  # Disable to avoid metric dependency
+        # metric_for_best_model="eval_loss",  # Disabled - no best model selection
+        # greater_is_better=False,  # Disabled
         logging_steps=25,
         report_to="none",
         dataloader_num_workers=2,