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

 

아래와 같은 에러시 

 

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)