Community

yolov5 훈련 시간

현재 제가 사용하고 있는 로컬의 사양은 i711700, 32g ram, rtx 4070 vram12g입니다. 코랩에서 제공하는 gpu(t4)보다 상위의 사양으로 알고 있습니다. 간단한 이미지 분류에서도 훈련 시간 차이에서 확인하였습니다. 본론으로 들어가 현재 저는 1200장의 데이터 중 300장을 train test split하여 yolov5모델을 학습하고 검증 중입니다. 여기에서 학습을 하기 위한, yolov5의 train.py를 실행할 때, 똑같은 데이터와 코드로 colab에서는 300장을 훈련시키는데에 10분 채 걸리지 않는데, 저의 로컬에서는 7시간이 지나도 훈련 중입니다. 하지만 300장이 아닌 100장으로 로컬에서 훈련 시켰을 땐 90초 이내로 훈련이 완료되었습니다. 이 문제에 대해 설명해주시면 감사하겠습니다.

Loading...

알림

알림이 없습니다