RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

ResNet 코드 실행 중 아래 그림과 같은 오류가 발생했다.

error 캡쳐본

torch에서 GPU할당을 받기 위해 사용하는 코드를 임의로 주석 처리해서 발생한 에러였다. 

이 문제는 모델은 GPU에 있지만, 데이터는 CPU에 있기 때문에 나오는 문제이다. 따라서 입력 tensor를 GPU로 보내도록 주석 해제를 하면 에러가 해결된다.

error 원인

주석을 풀고 GPU할당을 하도록 하여  .cuda()에 대한 오류를 해결하였다.

error 해결

정상적으로 작동하는 모습

 

---