본문 바로가기

Data-science/deep learning

[pytorch] Expected cuda got cpu, 혹은 타입 에러 발생시

728x90

내가 주로 사용하는 함수는 

torch.FloatTensor()

torch.LongTensor()

인데 이게 안 먹는 경우가 있다.

그럴 경우

.float()

.long()

으로 하면 된다.