반응형
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' 카테고리의 다른 글
| 3D points data scatter plot (2) | 2024.10.08 |
|---|---|
| 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 |