인공지능/이미지 인식
Error: cuDNN isn't found FWD algo for convolution. 해결 방법
미니언즈 닮은 개발자
2022. 1. 28. 10:45
GPU 하드웨어 성능이 떨어져서 발생하는 문제이다.
따라서, cfg 파일을 아래와 같이 변경하면 오류를 해결할 수 있다.
[net]
batch=64
subdivisions=64
width=64
height=64
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1
learning_rate=0.001
burn_in=1000
max_batches = 2200
policy=steps
steps=4800,5400
scales=.1,.1