336x280(권장), 300x250(권장), 250x250, 200x200 크기의 광고 코드만 넣을 수 있습니다.
checkpoint = torch.load(ckpt_dir, map_location=lambda storage, loc: storage)
self.model.load_state_dict(checkpoint['state_dict'])또는
device = torch.device('cpu')
checkpoint = torch.load(ckpt_dir, map_location=device)
self.model.load_state_dict(checkpoint['state_dict'])로 불러올 수 있다.
'AI > Tip' 카테고리의 다른 글
| zsh: command not found: python3 (0) | 2020.02.17 |
|---|---|
| [sklearn] train, test set 분할 (0) | 2020.02.16 |
| tensor 차원 늘리기 & channel 추가 (0) | 2020.02.06 |
| docker ssh 설치 및 sublimetext sftp & filezilla 연결하기 (0) | 2020.01.16 |
| zsh: command not found: brew (0) | 2020.01.16 |
