Data-science/deep learning
[pytorch] Expected cuda got cpu, 혹은 타입 에러 발생시
study&grow
2022. 3. 20. 15:27
728x90
내가 주로 사용하는 함수는
torch.FloatTensor()
torch.LongTensor()
인데 이게 안 먹는 경우가 있다.
그럴 경우
.float()
.long()
으로 하면 된다.