[Hugging Face] Trainer.train() 사용 시 학습 이어서 하기

trainer.train(resume_from_checkpoint=True)

 

단 위 코드의 한 가지 문제점은 처음 저 코드를 실행하면 아래와 같은 에러가 뜬다는 거다. 일단 try-except문으로 코드를 수정해뒀는데 임시 해결책인 것 같기도 해서 살짝 찝찝하긴 하다.

ValueError: No valid checkpoint found in output directory (TrainingArguments에 지정한 output 디렉토리명)