torch 모델 불러오기

GPU 에서 작업하고 저장한 모델이다. 하지만 GPU가 없는 컴퓨터에서 작업을 하기 위해서 모델을 불러오면 오류가 발생한다. 오류 메세지에 어떻게 하라는 설명이 나오기는 하니까 어렵지 않게 해결할 수 있다. map_location = ‘cpu’를 명시하면서 불러오면 된다.

model = torch.load('model.pt', map_location='cpu')