반응형
mps 옵션을 넣으면 m1 환경에서 훈련을 이용할 수 있다.
문제는 가벼운 모델 학습 정도를 가정한 것인지 yolov8s 모델을 학습시킬 땐 메모리가 꽉 차서 마우스도 못움직이고 맥북이 멈췄다가 강제종료 됐다.
batch size를 줄이고 모델도 yolov8n으로 낮추고 했는데도 1 epoch에 걸리는 시간이 심상치 않다.
이미지 데이터 2000장 분량이지만 역시 일반적인 GPU 환경에 비하면 너무 심하다.
나중에 좋은 최신 맥북 프로를 맞추면 가능할지 모르지만 어쨌든 지금은 내 맥북에서 훈련은 꿈도 꾸지 말아야겠다.
yolo 모델의 경우 학습을 200~300에폭을 돌려야 효과적으로 학습한다는데 현재 colab은 그렇게 돌리면 도중에 런타임 연결이 끊겨버려서 50에폭씩 학습시키며 학습파일을 드라이브에 저장하고 그것을 resume 옵션으로 이어서 학습하는 방식을 써야할 것 같다.
반응형
'프로그래밍 > 데이터 사이언스 공부' 카테고리의 다른 글
Yolov8s 모델에 바로 resume 옵션을 사용하는건 에러가 나온다 (0) | 2023.10.18 |
---|---|
Windows CUDA 설치 과정 (0) | 2023.10.18 |
중단된 Yolov8 모델 훈련을 이어서할 때 필수인 resume 옵션 (1) | 2023.10.17 |
YOLOv8 차량 추적, 집계 실습 (0) | 2023.09.23 |
YOLOv8 Object Tracking 실습 (0) | 2023.09.23 |