From 9097cd7ba3449d2ec0e567bf7b4178e7ebdbd281 Mon Sep 17 00:00:00 2001
From: Parsa <parsa.rahimi.n@gmail.com>
Date: Sat, 10 Jun 2023 18:36:19 +0200
Subject: [PATCH] Update losses.py

Incorrect `scale` parameter name for ArcFace class.
---
 recognition/arcface_torch/losses.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/recognition/arcface_torch/losses.py b/recognition/arcface_torch/losses.py
index e0b4585..7d54414 100644
--- a/recognition/arcface_torch/losses.py
+++ b/recognition/arcface_torch/losses.py
@@ -62,7 +62,7 @@ class ArcFace(torch.nn.Module):
     """
     def __init__(self, s=64.0, margin=0.5):
         super(ArcFace, self).__init__()
-        self.scale = s
+        self.s = s
         self.margin = margin
         self.cos_m = math.cos(margin)
         self.sin_m = math.sin(margin)
@@ -81,7 +81,7 @@ class ArcFace(torch.nn.Module):
             final_target_logit = target_logit + self.margin
             logits[index, labels[index].view(-1)] = final_target_logit
             logits.cos_()
-        logits = logits * self.s        
+        logits = logits * self.s   
         return logits