반응형 pytorch1 [PyTorch] torch.load 시, 학습 환경과 달라 모델 로딩이 되지 않는 에러 (map_location으로 해결) 에러 발생 내용 Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU. CUDA 환경에서 학습한 모델을 CPU로 가져와서 모델을 사용할 때 load_state_dict 함수를 사용해서 모델을 로딩한다. 이 때, load_state_dict(torch.load(model_path)) 위의 에러 메시지와 함께 모델이 정상적으로 올라가지 않을 때가 있다. 이 때 에러 .. 2022. 6. 24. 이전 1 다음 반응형