Alex/python.traduction
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

optimisation de la relance

Browse Source
This commit is contained in:
Alex 2026-01-14 17:59:36 +01:00
parent adca297850
commit 8d2e5ac021
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Finetunning/finetunning.py
View File

@ -88,7 +88,7 @@ print("LoRA adapters attached.")
print(f"{80 * '_'}\n[5/7] Loading dataset from JSON file...") print(f"{80 * '_'}\n[5/7] Loading dataset from JSON file...")
dataset = load_dataset( dataset = load_dataset(
"json", "json",
data_files="traductions.json" data_files="paires.json"
) )
print(f"Dataset loaded with {len(dataset['train'])} samples.") print(f"Dataset loaded with {len(dataset['train'])} samples.")
@ -147,7 +147,10 @@ print("Trainer initialized.")
# Train # Train
# ---------------------------- # ----------------------------
print(f"{80 * '_'}\n[7/7] Starting training...") 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.") print("Training completed successfully.")
# ---------------------------- # ----------------------------