모델 파일 로드시 에러

마이클의 AI 연구소·2022년 10월 25일
0
post-thumbnail

오류 발생

파이토치 모델 파일을 로드할 때, 다음과 같은 에러가 발생했다.

  File "/home/stans/anaconda3/envs/keti-gui-3d/lib/python3.9/site-packages/torch/serialization.py", line 920, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.

원인분석 및 해결방안

알고보니 git-lfs로 모델 파일을 운용중이었는데, 해당 저장소를 클론한 개발서버의 git에 lfs 관련 기능이 포함되지 않은 버전이었는지 모델파일이 lfs meta그대로 다운로드 되었던 것이다.

다음과 같이 설치 한 결과 정상적으로 수행되었습니다.

$ curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash
$ sudo apt-get install git-lfs
profile
늘 성장을 꿈꾸는 자들을 위한 블로그입니다.

0개의 댓글