다른 모델을 내 모델로 불러올 때 Unexpected key(s) in state_dic 문제

🤦‍♂️ 학습한 모델을 파인튜닝을 하기 위해 torch.load를 이용해 다시 불러오려고 했는데, 아래와 같은 에러가 발생했다.

어떤 부분이 다른가 했더니 파인튜닝 시에는 아래 2번 부분을 없애려고 했다.

해결방법

아래와 같이 인자에 strict를 False 로 설정하여 일치하지 않는 키들을 무시하도록 설정한다.

기존

model.reformer.load_state_dict(torch.load(args.checkpoint, map_location=device))

수정

model.reformer.load_state_dict(torch.load(args.checkpoint, map_location=device),strict=False)