[torch] RuntimeError("{} is a zip archive 에러 해결 방법

에러

RuntimeError("{} is a zip archive (did you mean to use torch.jit.load()?)".format(f.name)) when loading model weights

문제 원인 :

학습은 torch 1.6 이상으로 하고

인퍼런스는 torch 1.6 버전 이하로 할 경우 위와 같은 에러가 발생한다.

해결 방법 : 

학습시 모델을 저장할 때 다음과 같은 옵션을 추가하여 저장해준다.

torch.save(model.state_dict(), path, _use_new_zipfile_serialization=False)