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

aam

parent 0f3b874a
......@@ -250,7 +250,7 @@ class ArcMarginProduct(torch.nn.Module):
self.th = math.cos(math.pi - self.m)
self.mm = math.sin(math.pi - self.m) * self.m
def __change_params(self, s=None, m=None):
def change_params(self, s=None, m=None):
if s is None:
s = self.s
if m is None:
......
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