(self, ckpt_path)
| 33 | self.device = device |
| 34 | |
| 35 | def load_ckpt(self, ckpt_path): |
| 36 | checkpoint_dict = torch.load(ckpt_path, map_location=torch.device(self.device)) |
| 37 | a, b = self.model.load_state_dict(checkpoint_dict['model'], strict=False) |
| 38 | print("Loaded checkpoint '{}'".format(ckpt_path)) |
| 39 | print('missing/unexpected keys:', a, b) |
| 40 | |
| 41 | |
| 42 | class BaseSpeakerTTS(OpenVoiceBaseClass): |