Commit 98fdd2d7 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

arcface

parent 13062045
......@@ -528,7 +528,7 @@ class Xtractor(torch.nn.Module):
self.after_speaker_embedding = ArcFace(embedding_size=input_size,
classnum=self.speaker_number,
s=64.,
margin=0.5)
m=0.5)
self.after_speaker_embedding_weight_decay = cfg["after_embedding"]["weight_decay"]
......
Supports Markdown
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