Update train.py

train.py

@@ -47,20 +47,17 @@ print("Dataset successfully split and tokenized.")
 
 #  Define training arguments
 training_args = TrainingArguments(
-    output_dir="./results",
-    evaluation_strategy="epoch",
-    save_strategy="epoch",
-    learning_rate=5e-6,  # Reduce from 5e-5 to 5e-6
-    per_device_train_batch_size=8,  # Keep batch size reasonable
+    output_dir="/tmp/results",  # Use /tmp/ to avoid permission errors
+    per_device_train_batch_size=8,
     per_device_eval_batch_size=8,
-    num_train_epochs=3,
-    weight_decay=0.01,
-    logging_dir="./logs",
-    logging_steps=10,
+    evaluation_strategy="steps",
+    save_steps=500,
+    eval_steps=500,
+    logging_dir="/tmp/logs",  # Avoid writing to restricted directories
+    logging_steps=100,
+    save_total_limit=2,
     fp16=True
 )
-
-
 # Set up Trainer
 trainer = Trainer(
     model=model,