Upload train_code_reasoning.py

train_code_reasoning.py

@@ -166,7 +166,7 @@ training_args = SFTConfig(
     per_device_train_batch_size=4,
     gradient_accumulation_steps=4,
     learning_rate=5e-5,
-    warmup_ratio=0.03,
+    warmup_steps=300,
     lr_scheduler_type="cosine",
     bf16=True,
     gradient_checkpointing=True,
@@ -181,7 +181,6 @@ training_args = SFTConfig(
     report_to=["trackio"],
     seed=42,
     hub_strategy="checkpoint",
-    hub_always_push=True,
 )
 
 print("\nInitializing SFTTrainer...")
@@ -189,7 +188,7 @@ trainer = SFTTrainer(
     model=MODEL_ID,
     train_dataset=train_dataset,
     args=training_args,
-    tokenizer=tokenizer,
+    processing_class=tokenizer,
     max_seq_length=2048,
 )