Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add load_chekpoint to speaker encoder

This commit is contained in:
Eren Gölge 2021-04-21 13:09:04 +02:00
parent 1229ccbf07
commit 2da81f5bb6
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
TTS/speaker_encoder/model.py
View File

@ -108,3 +108,11 @@ class SpeakerEncoder(nn.Module):
else: else:
embed[cur_iter <= num_iters, :] += self.inference(frames[cur_iter <= num_iters, :, :]) embed[cur_iter <= num_iters, :] += self.inference(frames[cur_iter <= num_iters, :, :])
return embed / num_iters return embed / num_iters
# pylint: disable=unused-argument, redefined-builtin
def load_checkpoint(self, config: dict, checkpoint_path: str, eval: bool = False):
state = torch.load(checkpoint_path, map_location=torch.device("cpu"))
self.load_state_dict(state["model"])
if eval:
self.eval()
assert not self.training