change angle_tau to 20.0

SeanLee97 · Feb 28, 2024 · 05a2c59 · 05a2c59
1 parent c252268
commit 05a2c59
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/angle_emb/train_cli.py b/angle_emb/train_cli.py
@@ -43,8 +43,8 @@
                     help='Specify w2 (ibn), default 1.0')
 parser.add_argument('--w3', type=float, default=1.0,
                     help='Specify w3 (angle), default 1.0')
-parser.add_argument('--angle_tau', type=float, default=1.0,
-                    help='Specify angle_tau, default 1.0')
+parser.add_argument('--angle_tau', type=float, default=20.0,
+                    help='Specify angle_tau, default 20.0')
 parser.add_argument('--cosine_tau', type=float, default=20.0,
                     help='Specify cosine_tau, defaut 20.0')
 parser.add_argument('--ibn_tau', type=float, default=20.0,