GPU
-
[이거 어떡하지] optim in step: expected device cpu but got device cuda:0Pytorch 2021. 6. 11. 23:08
증상: optim.step() - RuntimeError: expected device cpu but got device cuda:0 1. pytorch에서 코딩하는데 난 model도 cuda에 올렸고 (Model().cuda() 혹은 Model().to(device)를 사용) 2. torch.load({CHECKPOINT_PATH})로 optimzer의 state를 불러왔는데 아래와 같이 optimizer의 step()에서 이러한 증상이 나타납니다. 왜 그럴까요? 해결 방법: 1] optim_state = torch.load({CHECKPOINT_PATH})["optimizer"] 2] optim.load_state_dict(optim_state) 3] for state in optim.state.va..