728x90
반응형
발생 상황
cudstom data로 특정 network에서 학습시키고자 할때 발생한 에러
해결 방법
num_classes를 체크하고 그를 수정. 필자의 경우 9개의 classes를 예측해야했기 때문에 9개로 수정
728x90
반응형
발생 상황
cudstom data로 특정 network에서 학습시키고자 할때 발생한 에러
해결 방법
num_classes를 체크하고 그를 수정. 필자의 경우 9개의 classes를 예측해야했기 때문에 9개로 수정