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

cleaning

parent f95e0411
......@@ -516,12 +516,12 @@ class Xtractor(torch.nn.Module):
("lin_be", torch.nn.Linear(in_features = 5120,
out_features = self.embedding_size,
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,
# 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
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