Update training_config.py

training_config.py

@@ -3,7 +3,7 @@ from transformers import TrainingArguments
 training_args = TrainingArguments(
     output_dir="./results",
     num_train_epochs=3,
-    per_device_train_batch_size=8,  # Optimized for T4 GPU
+    per_device_train_batch_size=8,
     per_device_eval_batch_size=8,
     warmup_steps=500,
     weight_decay=0.01,
@@ -16,6 +16,9 @@ training_args = TrainingArguments(
     save_total_limit=3,
     load_best_model_at_end=True,
     learning_rate=5e-5,
-    fp16=True,  # Using GPU features
-    gradient_checkpointing=True
+    fp16=True,
+    gradient_checkpointing=True,
+    # Remove CPU-only settings
+    no_cuda=False,  # Allow GPU usage
+    use_cpu=False   # Allow GPU usage
 )