From 892c3c36232a6a48febfc67f2f0f72c3a425b77f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 3 Mar 2021 15:37:38 +0100 Subject: [PATCH] use torch for AngleProtoLoss --- TTS/speaker_encoder/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/speaker_encoder/losses.py b/TTS/speaker_encoder/losses.py index 35ff73fa..9e7bc265 100644 --- a/TTS/speaker_encoder/losses.py +++ b/TTS/speaker_encoder/losses.py @@ -155,6 +155,6 @@ class AngleProtoLoss(nn.Module): cos_sim_matrix = F.cosine_similarity(out_positive.unsqueeze(-1).expand(-1, -1, num_speakers), out_anchor.unsqueeze(-1).expand(-1, -1, num_speakers).transpose(0, 2)) torch.clamp(self.w, 1e-6) cos_sim_matrix = cos_sim_matrix * self.w + self.b - label = torch.from_numpy(np.asarray(range(0, num_speakers))).to(cos_sim_matrix.device) + label = torch.arange(num_speakers).to(cos_sim_matrix.device) L = self.criterion(cos_sim_matrix, label) return L