浏览代码

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.")

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

正在加载...
取消
保存