728x90
반응형
nn.Conv2d(1, 3, 3, stride=2) # (입력 채널 수, 출력 채널 수, 필터 크기, 스트라이드)

나의 경우에는 이런식으로 Conv2d Layer를 쌓아서 에러가 났다.

이 에러가 난 이유는 입력으로 주어지는 이미지가 컬러인데 입력 채널 수를 1로 설정했기 때문..

 

딥러닝 코딩 초보라서 하나하나 맨땅에 헤딩하면서 배워가는 중,,,

728x90
반응형
  • 네이버 블러그 공유하기
  • 네이버 밴드에 공유하기
  • 페이스북 공유하기
  • 카카오스토리 공유하기