Python
현재 모델의 state_dict()가 갖는 key값만 저장된 state_dict()로 부터 불러오기
ddokkddokk
2023. 6. 2. 14:21
반응형
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)