-
질문 & 답변
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
20.11.14 12:29 작성 조회수 500
0
CGAN 코드의 Line 11 번에서 아래와 같은 오류가 나타납니다.
"for i, (images, labels) in enumerate(data_loader): "에 해당하는 부분입니다.
dataset 의 차원이 일치 않는 것 같은데, 코드 수정을 부탁합니다.
답변을 작성해보세요.
0
코코
지식공유자2020.11.17
안녕하세요. 님
답변이 늦어서 죄송합니다. DCGAN에 대한 에러도 질문을 해주셨는데 여기서 답변을 같이 해드리도록 하겠습니다.
우선 제가 올려드린 코드는 다시 확인해 보았는데 정상적으로 잘 작동하는 것을 확인하였습니다.
올려주신 캡쳐화면으로는 원인을 알기는 쉽지 않은 것 같은데
추정하기로는 저와 환경이 달라서 그런게 아닌가 싶습니다.
이전에 질문을 올려주신바와 같이 Type error는
에러 메세지에 따라서 다음과 같이 수정해 주시면 됩니다.
output.type(torch.LongTensor)
output.type(torch.FloatTensor)
이에 관해서는 아래링크를 참고해주시면 감사하겠습니다.
https://stackoverflow.com/questions/60440292/runtimeerror-expected-scalar-type-long-but-found-float
더불어 제 Python 환경은 3.6.5이고 Torch는 1.0.0 입니다.
답변 1