Переглянути джерело

optimisation de la relance

main
Alex 1 тиждень тому
джерело
коміт
8d2e5ac021
1 змінених файлів з 5 додано та 2 видалено
  1. 5
    2
      Finetunning/finetunning.py

+ 5
- 2
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.")

# ----------------------------

Завантаження…
Відмінити
Зберегти