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

add BN in before embedding

parent 817431b6
......@@ -514,7 +514,8 @@ class Xtractor(torch.nn.Module):
self.before_speaker_embedding = torch.nn.Sequential(OrderedDict([
("lin_be", torch.nn.Linear(in_features = 5120,
out_features = self.embedding_size)),
out_features = self.embedding_size,
bias=False)),
("bn_be", torch.nn.BatchNorm1d(self.embedding_size))
]))
......
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