ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-16 05:27:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix combined margin

Browse Source
This commit is contained in:
Jia Guo
2018-03-26 17:27:06 +08:00
parent 8bfb6f26e8
commit b939ef5ebc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/train_softmax.py
View File

@@ -249,7 +249,7 @@ def get_symbol(args, arg_params, aux_params):
zy = mx.sym.pick(fc7, gt_label, axis=1)
cos_t = zy/s
t = mx.sym.arccos(cos_t)
if args.margin_a>0.0:
if args.margin_a!=1.0:
t = t*args.margin_a
if args.margin_m>0.0:
t = t+args.margin_m