Update spherefacer.py

ydwen · Apr 28, 2022 · c6ceee5 · c6ceee5
1 parent b571c10
commit c6ceee5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model/head/spherefacer.py b/model/head/spherefacer.py
@@ -29,6 +29,7 @@ def get_d_theta(cos_theta, y, magn_type, m):
 
     return d_theta
 
+# Warning: Both SphereFaceR_N and SphereFaceR_S are needed to be tested before using.
 
 class SphereFaceR_N(nn.Module):
     """ reference: <SphereFace Revived: Unifying Hyperspherical Face Recognition>