Commit f404bddb authored by Anthony Larcher's avatar Anthony Larcher
Browse files

cleaning

parent f95e0411
...@@ -516,12 +516,12 @@ class Xtractor(torch.nn.Module): ...@@ -516,12 +516,12 @@ class Xtractor(torch.nn.Module):
("lin_be", torch.nn.Linear(in_features = 5120, ("lin_be", torch.nn.Linear(in_features = 5120,
out_features = self.embedding_size, out_features = self.embedding_size,
bias=False)), bias=False)),
("bn_be", torch.nn.BatchNorm1d(self.embedding_size)) #("bn_be", torch.nn.BatchNorm1d(self.embedding_size))
])) ]))
#self.before_speaker_embedding = torch.nn.Linear(in_features = 5120, #self.before_speaker_embedding = torch.nn.Linear(in_features = 5120,
# out_features = self.embedding_size) # out_features = self.embedding_size)
self.stat_pooling = AttentivePooling(256, 80, global_context=True) self.stat_pooling = AttentivePooling(self.embedding_size, 80, global_context=True)
self.loss = loss self.loss = loss
if self.loss == "aam": if self.loss == "aam":
......
Markdown is supported
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