반응형
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)
'Python' 카테고리의 다른 글
Scatter plot with subfigures (0) | 2023.11.13 |
---|---|
PIL.image 라이브러리 사용법 (0) | 2023.05.31 |
Change working directory to current location (0) | 2023.05.18 |
nuScenes 데이터셋의 이해 (0) | 2023.05.12 |
[matplotlib] Creating multiple figures in one plot (0) | 2023.05.08 |