Yolov8s 모델에 바로 resume 옵션을 사용하는건 에러가 나온다 docs에선 커스텀 모델에 대해서 resume을 하라고 나왔는데 이미 pretrained 된 yolov8s 모델에 resume 옵션을 쓰면 car 클래스는 yolov8s가 coco데이터셋으로 미리 학습해놨을테니 좋지 않을까 해서 시도해봤는데 에러가 나온다. 이전 가중치를 그대로 이어서 하는거라 그런가보다 프로그래밍/데이터 사이언스 공부 2023.10.18
Windows CUDA 설치 과정 https://teang1995.tistory.com/17 [ 개발 환경 설정 ] Windows 10 에 CUDA를 설치해보자! 아! 입사 첫 주라 설정해줄 것이 정말 징글징글하게 많다! 전 게시물이 conda + vscode 였다면, 이번엔 CUDA + Torch다! conda 설정 및 python 개발 환경 구축 : link 1. 내 GPU 확인 어렵지 않다. 우선 시작 메뉴 teang1995.tistory.com 위 블로그의 과정 그대로 진행한다. https://kjk92.tistory.com/82 CUDA 설치 및 설치 실패 대처 목적 1) CUDA 설치 및 CUDA 설치 확인 2) CUDA 설치 오류 대처 1) CUDA 설치 및 CUDA 설치 확인 CUDA 11.0버전을 설치했습니다. Visua.. 프로그래밍/데이터 사이언스 공부 2023.10.18
yolov8를 mac m1칩으로 훈련 mps 옵션을 넣으면 m1 환경에서 훈련을 이용할 수 있다. 문제는 가벼운 모델 학습 정도를 가정한 것인지 yolov8s 모델을 학습시킬 땐 메모리가 꽉 차서 마우스도 못움직이고 맥북이 멈췄다가 강제종료 됐다. batch size를 줄이고 모델도 yolov8n으로 낮추고 했는데도 1 epoch에 걸리는 시간이 심상치 않다. 이미지 데이터 2000장 분량이지만 역시 일반적인 GPU 환경에 비하면 너무 심하다. 나중에 좋은 최신 맥북 프로를 맞추면 가능할지 모르지만 어쨌든 지금은 내 맥북에서 훈련은 꿈도 꾸지 말아야겠다. yolo 모델의 경우 학습을 200~300에폭을 돌려야 효과적으로 학습한다는데 현재 colab은 그렇게 돌리면 도중에 런타임 연결이 끊겨버려서 50에폭씩 학습시키며 학습파일을 드라이브에 저.. 프로그래밍/데이터 사이언스 공부 2023.10.18
