Skip to content
Snippets Groups Projects
Commit c8059a42 authored by Ketan Kotwal's avatar Ketan Kotwal
Browse files

concat scores for intra-demog loss

parent fa69312d
No related branches found
No related tags found
No related merge requests found
...@@ -84,7 +84,7 @@ class IntraDemogLoss(nn.Module): ...@@ -84,7 +84,7 @@ class IntraDemogLoss(nn.Module):
cat_scores = torch.cat((genuine_scores, impostor_scores)) cat_scores = torch.cat((genuine_scores, impostor_scores))
cat_labels = torch.cat((genuine_labels, imposter_labels)) cat_labels = torch.cat((genuine_labels, imposter_labels))
output = self.loss_fn(new_scores[:, 0], new_labels) output = self.loss_fn(cat_scores[:, 0], cat_labels)
return output return output
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment