Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same
|2021. 6. 24. 17:30
728x90
반응형
아래와 같은 에러시
RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same
x = x.to(device).float()
.float() 사용하여 해결한다.
참고자료 : https://m.blog.naver.com/PostView.naver?isHttpsRedirect=true&blogId=chrhdhkd&logNo=221469768189
pytorch : RuntimeError 발생시 해결방법(Input type and weight type should be the same)
발생한 에러 : RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.F...
blog.naver.com
728x90
반응형
'삽질 기록' 카테고리의 다른 글
| Image Classification 시 분류 결과 한쪽에 치우쳐져서 나오는 현상 (0) | 2021.07.01 |
|---|---|
| Linux 환경에서 OpenCV의 VideoCapture 사용 시 Webcam이 죽어도 읽히지 않을 때 (0) | 2021.06.25 |
| Ubuntu 18.04 환경에서 VNC Server 설정했을 때 회색 화면 나타나는 문제 (0) | 2021.06.11 |
| Pytorch 에서의 TopK error (k not in range for dimension) (0) | 2021.05.10 |
| VNC viewer 에서 Pycharm 실행 시 DISPLAY 문제 (0) | 2021.05.10 |