YOLOv5 이어서 학습하기 --resume 사용
·
ML & DL
YOLOv5를 학습하다가 예기지 못하게 중간에 중단 현상이 나타났다.(사실 전적으로 본인의 잘못😭) 그래서 YOLOv5 학습 도중 중단되었을 때 학습을 이어나가는 방법을 정리하고자 한다. YOLOv5 이어서 학습하기--resume을 사용하면 된다! YOLO 처럼 유명한 모델은 대부분 weight를 기록하고 저장하는 코드가 굉장히 잘 짜여 있다. 그렇기 때문에 그런 모델을 사용할 경우에 우리는 있는 것을 잘 찾아서 사용하기만 하면 된다! 원래 다음과 같은 명령어를 실행해서 epoch는 100으로 설정하고 훈련시키고 있었다.python train.py --img 640 --batch 64 --epochs 100 --data data/data.yaml --cfg models/yolov5s.yaml --..
