Commit 6430129d authored by Anthony Larcher's avatar Anthony Larcher
Browse files

update resnet34

parent d8ac6322
......@@ -406,23 +406,21 @@ class Xtractor(torch.nn.Module):
elif model_archi == "resnet34":
self.preprocessor = MfccFrontEnd()
self.preprocessor = MelSpecFrontEnd(n_mels=80)
self.sequence_network = PreResNet34()
self.before_speaker_embedding = torch.nn.Linear(in_features = 5120,
out_features = 256)
self.before_speaker_embedding = torch.nn.Linear(in_features=5120,
out_features=self.embedding_size)
self.stat_pooling = MeanStdPooling()
self.stat_pooling_weight_decay = 0
self.embedding_size = 256
self.loss = "aam"
self.after_speaker_embedding = ArcMarginProduct(256,
self.after_speaker_embedding = ArcMarginProduct(self.embedding_size,
int(self.speaker_number),
s = 30.0,
m = 0.20,
easy_margin = True)
easy_margin = False)
self.preprocessor_weight_decay = 0.000
self.sequence_network_weight_decay = 0.000
......
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