728x90
반응형
nn.Conv2d(1, 3, 3, stride=2) # (입력 채널 수, 출력 채널 수, 필터 크기, 스트라이드)
나의 경우에는 이런식으로 Conv2d Layer를 쌓아서 에러가 났다.
이 에러가 난 이유는 입력으로 주어지는 이미지가 컬러인데 입력 채널 수를 1로 설정했기 때문..
딥러닝 코딩 초보라서 하나하나 맨땅에 헤딩하면서 배워가는 중,,,
728x90
반응형
최근댓글