RuntimeError: result type Float can't be cast to the desired output type Long
·
Error Note
- 전체 에러 문구 RuntimeError: result type Float can't be cast to the desired output type Long => output 텐서와 target 텐서의 데이터 타입이 같지 않을 때 발생하는 에러입니다. - 해결 방법 저의 경우에는 output 텐서는 float32 타입, target 텐서는 int64 타입이었습니다. 두 텐서의 데이터 타입을 동일하게 맞춰주었습니다. 따라서, target텐서를 float 타입으로 변경해주었습니다. labels => labels.float() self.loss(logits, labels.view([-1,1]).float(), alpha=0.75, reduction='mean')