소스 검색

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

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

Loading…
취소
저장