오답노트

[yolov5] RuntimeError : unable to find a valid cuDNN convolution 본문

머신러닝, 딥러닝

[yolov5] RuntimeError : unable to find a valid cuDNN convolution

장비 정 2021. 11. 24. 10:03

직접 라벨링한 학습 데이터를 토대로 train 을 시킬 때 해당 오류 발생.

 

batch size 를 기존 32 에서 16 으로 줄이니 해결 되었다.