본문 바로가기

Python

현재 모델의 state_dict()가 갖는 key값만 저장된 state_dict()로 부터 불러오기

반응형
pretrained_dict = {k: v for k, v in checkpoint['scratch_state_dict'].items() if k in self.scratch.state_dict()}
self.scratch.load_state_dict(pretrained_dict)