728x90
RuntimeError: result type Float can't be cast to the desired output type Long
y의 텐서가 float의 형태가 아니라 생기는 오류였다....
dtype=torch.float32 추가하면 float형태로 반환된다!
728x90
'AI 공부 > 딥러닝' 카테고리의 다른 글
| TabNet 정리 (0) | 2022.12.12 |
|---|---|
| 평가 지표 (0) | 2022.11.07 |
| 딥러닝 (RNN,LSTM,GRU) (1) | 2022.10.03 |
| 딥러닝 (손실함수) (0) | 2022.10.02 |
| 딥러닝 (pytorch) (0) | 2022.10.02 |
댓글