chore: _

trainer.py

@@ -55,7 +55,8 @@ class CustomTrainer:
             output_dir=self.config.output_dir,
             per_device_train_batch_size=self.config.per_device_train_batch_size,
             gradient_accumulation_steps=self.config.gradient_accumulation_steps,
-            warmup_ratio=self.config.warmup_ratio,
+            # warmup_ratio=self.config.warmup_ratio,
+            warmup_steps=self.config.warmup_steps,
             max_grad_norm=self.config.max_grad_norm,
             num_train_epochs=self.config.num_train_epochs,
             learning_rate=self.config.learning_rate,