Commit 9385cb80 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

debug

parent f404bddb
......@@ -511,17 +511,9 @@ class Xtractor(torch.nn.Module):
n_mels=80)
self.sequence_network = PreHalfResNet34()
self.embedding_size = 256
self.before_speaker_embedding = torch.nn.Sequential(OrderedDict([
("lin_be", torch.nn.Linear(in_features = 5120,
out_features = self.embedding_size,
bias=False)),
#("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(self.embedding_size, 80, global_context=True)
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.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