RuntimeError: result type Float can't be cast to the desired output type Long

반응형

- 전체 에러 문구

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')