Create core/trainer.py

core/trainer.py

@@ -0,0 +1,22 @@
+from transformers import Trainer, TrainingArguments
+
+def build_trainer(model, dataset, tokenizer, hub_model_id):
+    print("⚙️ Initialisation du Trainer...")
+    args = TrainingArguments(
+        output_dir="./output",
+        per_device_train_batch_size=2,
+        num_train_epochs=1,
+        save_steps=500,
+        learning_rate=2e-5,
+        logging_dir="./logs",
+        push_to_hub=True,
+        hub_model_id=hub_model_id
+    )
+
+    trainer = Trainer(
+        model=model,
+        args=args,
+        train_dataset=dataset,
+        tokenizer=tokenizer
+    )
+    return trainer