[Deep Learning] [Pytorch] ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1]) – model.eval()로 해결 ! (tistory.com)
이걸 좀 오래 잡고 있었는데, model.eval() 을 통해서 해결하면 된다고 한다…
Share via:
이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다
댓글 *
이름 *
이메일 *
웹사이트
다음 번 댓글 작성을 위해 이 브라우저에 이름, 이메일, 그리고 웹사이트를 저장합니다.
답글 남기기