use torch for AngleProtoLoss

coqui-ai · Mar 30, 2021 · 892c3c3 · 892c3c3
1 parent 2b3e12e
commit 892c3c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TTS/speaker_encoder/losses.py b/TTS/speaker_encoder/losses.py
@@ -155,6 +155,6 @@ def forward(self, x):
         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