프로그래밍/데이터 사이언스 공부

중단된 Yolov8 모델 훈련을 이어서할 때 필수인 resume 옵션

이자다 2023. 10. 17. 22:28
반응형

 

 

yolov8 모델을 훈련시킬 때 의도치 않게 학습이 중단됐다면 resume 옵션을 사용하여 훈련을 재개할 수 있다.

 

이미 지정된 에폭만큼 반복하여 훈련을 완료한 모델에게 사용하면 이미 훈련 완료한 모델이라며 에러 메시지를 출력한다.

 

그리고 yolov8에서 기본적으로 제공하는, coco 데이터셋으로 pretrained된 모델에 resume 옵션을 사용하면 클래스 개수 혹은 사용하는 데이터셋이 coco와는 달라서인지 상술한 오류가 아닌 다른 오류가 출력된다.

반응형