티스토리 뷰

Computer/이게 왜 안되지?

[이왜안?] [RuntimeError]: Expected floating point type for target with class probabilities, got Long

벼랑끝과학자 2023. 7. 10. 18:06

torch의 nn module 내부의 loss function을 이용해서 loss를 구할 때에는 반드시 input의 type을 float으로 수정해준다.

# 이하 코드는 input의 type이 int(long tensor)였기 때문에 이러한 에러가 난다.

따라서 input에 .float()를 붙여서 float type으로 바꿔주면 문제없이 실행이 된다.

저작자표시

'Computer > 이게 왜 안되지?' 카테고리의 다른 글

[이왜안?] RuntimeError: Mask shape should match input shape; transformer_mask is not supported in the fallback case  (0) 2023.08.03
[이왜안?] failed to run cuBLAS routine: CUBLAS_STATUS_EXECUTION_FAILED  (0) 2023.07.26
[이왜안?] RuntimeError: stack(): functions with out=... arguments don't support automatic differentiation, but one of the arguments requires grad.  (0) 2023.03.04
[이왜안?] Transformer 모델 내부 mask의 device를 잘 설정하자  (0) 2023.02.20
[이왜안?] nn.Embedding, IndexError: index out of range in self  (0) 2023.02.13
댓글