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

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

 

원인

- 입력 Tensor랑 너가 만든 모델(네트워크)가 각각 다른 곳에서 돌고 있다.

- 예를 들어 입력 Tensor는 CPU쪽 메모리에 있는데 모델(네트워크)은 GPU에서 돌고 있다...

- 상호 참조가 안되서 생기는 문제다.

 

해결방안

- 입력 Tensor를 GPU 메모리로 보내면 된다.

- 예를 들어 입력 Tensor의 변수명이 input_tensor라 하자

- 그럼 input_tensor를 Variable(input_tensor.cuda())로 바꾸면 깔끔하게 해결된다. (Variable을 쓰려면 from torch.autograd import Variable를 코드 맨 앞에 써야한다. )