RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same에러를 읽어보자
인풋 데이터의 타입은 cuda에 올라가 있는데, weight는 cuda에 안올라가 있네?
이거 두개를 맞춰줘야 한다.
GPU를 사용한다면 model 과 dataset을 to("cuda")를 써서 GPU로 올리자.
GPU를 사용안한다면 그냥 하면 되지 않을까
'즐거운 에러 대잔치' 카테고리의 다른 글
| skimage를 import 하면 커널이 자꾸 죽는 현상 (0) | 2023.11.06 |
|---|---|
| RTX 3090, 3080을 위한 pytorch 버전 (0) | 2023.10.27 |
| cv2 VideoWriter 동영상 저장이 안될 때, 화면이 안나올 때 (0) | 2023.10.18 |
| Failed to load OpenH264 library: openh264, Could not open codec libopenh264 (0) | 2023.10.18 |
| OSError: [Errno 28] No space left on device (2) | 2023.10.17 |