torch.load 시 _pickle.UnpicklingError: invalid load key 해결

 

저장해뒀던 모델(.pt 파일)을 불러오려고 할 때 다음과 같은 에러가 발생할 때가 있습니다.

 

  model = torch.load(weights, map_location=device)

  File "작업 경로\venv\lib\site-packages\torch\serialization.py", line 713, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)

  File "작업 경로\venv\lib\site-packages\torch\serialization.py", line 920, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)

_pickle.UnpicklingError: invalid load key, '{WHATEVER}'.

이건 그냥 load하려는 모델 파일에 오류가 생긴 거니까, 결국 다시 가져와야 합니다.
다운로드를 하거나 애초에 torch.save 하면서 모델을 저장하는 과정에서 오류가 없었는지 확인하고,
정상적인 모델 파일을 넣어주면 해결됩니다.