output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]

Question

CGAN 코드의 Line 11 번에서 아래와 같은 오류가 나타납니다. "for i, (images, labels) in enumerate(data_loader): "에 해당하는 부분입니다. dataset 의 차원이 일치 않는 것 같은데, 코드 수정을 부탁합니다.

coco · Answer

안녕하세요. autumnfe79 님 답변이 늦어서 죄송합니다. 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 입니다.