From 8d2e5ac021c6d90d8630cfcd772ed53fae2f5240 Mon Sep 17 00:00:00 2001 From: Alex Date: Wed, 14 Jan 2026 17:59:36 +0100 Subject: [PATCH] optimisation de la relance --- Finetunning/finetunning.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/Finetunning/finetunning.py b/Finetunning/finetunning.py index 4b9e80f..cf8f687 100644 --- a/Finetunning/finetunning.py +++ b/Finetunning/finetunning.py @@ -88,7 +88,7 @@ print("LoRA adapters attached.") print(f"{80 * '_'}\n[5/7] Loading dataset from JSON file...") dataset = load_dataset( "json", - data_files="traductions.json" + data_files="paires.json" ) print(f"Dataset loaded with {len(dataset['train'])} samples.") @@ -147,7 +147,10 @@ print("Trainer initialized.") # Train # ---------------------------- print(f"{80 * '_'}\n[7/7] Starting training...") -trainer.train(resume_from_checkpoint=True) +try: + trainer.train(resume_from_checkpoint=True) +except: + trainer.train() print("Training completed successfully.") # ----------------------------