오답노트
[yolov5] RuntimeError : unable to find a valid cuDNN convolution 본문
직접 라벨링한 학습 데이터를 토대로 train 을 시킬 때 해당 오류 발생.
batch size 를 기존 32 에서 16 으로 줄이니 해결 되었다.
'머신러닝, 딥러닝' 카테고리의 다른 글
[efficientDet] ZeroDivisionError: integer division or modulo by zero (0) | 2021.12.15 |
---|---|
[yolov5] RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)` (0) | 2021.11.26 |
칼만필터 (Kalman Filter) (0) | 2021.11.09 |
semantic segmentation (0) | 2021.07.07 |
FairMOT (수정 중) (0) | 2021.06.08 |