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

yolov8를 mac m1칩으로 훈련

이자다 2023. 10. 18. 01:06
반응형

mps 옵션을 넣으면 m1 환경에서 훈련을 이용할 수 있다.

 

문제는 가벼운 모델 학습 정도를 가정한 것인지 yolov8s 모델을 학습시킬 땐 메모리가 꽉 차서 마우스도 못움직이고 맥북이 멈췄다가 강제종료 됐다.

 

batch size를 줄이고 모델도 yolov8n으로 낮추고 했는데도 1 epoch에 걸리는 시간이 심상치 않다.

 

 

 

 

이미지 데이터 2000장 분량이지만 역시 일반적인 GPU 환경에 비하면 너무 심하다.

 

나중에 좋은 최신 맥북 프로를 맞추면 가능할지 모르지만 어쨌든 지금은 내 맥북에서 훈련은 꿈도 꾸지 말아야겠다.

 

yolo 모델의 경우 학습을 200~300에폭을 돌려야 효과적으로 학습한다는데 현재 colab은 그렇게 돌리면 도중에 런타임 연결이 끊겨버려서 50에폭씩 학습시키며 학습파일을 드라이브에 저장하고 그것을 resume 옵션으로 이어서 학습하는 방식을 써야할 것 같다.

 

 

반응형