Commit 24d726cf authored by Anthony Larcher's avatar Anthony Larcher
Browse files

version a 5% d'EER

parent e6661e62
......@@ -198,7 +198,7 @@ class XtractorHot(Xtractor):
seg_emb_4 = self.norm7(self.activation(self.seg_lin1(seg_emb_3)))
# No batch-normalisation after this layer
seg_emb_5 = self.seg_lin2(seg_emb_4)
result = torch.nn.functional.softmax(self.activation(seg_emb_5),dim=1)
result = self.activation(seg_emb_5)
return result
def LossFN(self, x, label):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment