Study/딥러닝
to have 1 channels, but got 3 channels instead | 코랩 에러 | Pytorch
nn.Conv2d(1, 3, 3, stride=2) # (입력 채널 수, 출력 채널 수, 필터 크기, 스트라이드) 나의 경우에는 이런식으로 Conv2d Layer를 쌓아서 에러가 났다. 이 에러가 난 이유는 입력으로 주어지는 이미지가 컬러인데 입력 채널 수를 1로 설정했기 때문.. 딥러닝 코딩 초보라서 하나하나 맨땅에 헤딩하면서 배워가는 중,,,
2022. 3. 14. 04:08
최근댓글