cleaning

f404bddb · Anthony Larcher · f95e0411 · f404bddb
Commit f404bddb authored Apr 26, 2021 by Anthony Larcher
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

nnet/xvector.py nnet/xvector.py +2 -2

No files found.
--- a/nnet/xvector.py
+++ b/nnet/xvector.py
@@ -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":