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

tiny

Browse Source
This commit is contained in:
Jia Guo
2018-01-21 18:02:36 +08:00
parent 9ca64a1406
commit bc7d833935
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/train_softmax.py
View File

@@ -54,7 +54,7 @@ class AccMetric(mx.metric.EvalMetric):
self.count+=1
if len(preds)==4:
if self.count%args.ctx_num==0:
mbatch = self.count/args.ctx_num
mbatch = self.count//args.ctx_num
if mbatch==1 or mbatch%args.margin_verbose==0:
a = preds[-2].asnumpy()[0]
b = preds[-1].asnumpy()[0]
@@ -279,7 +279,7 @@ def get_symbol(args, arg_params, aux_params):
cond_v = cos_t - threshold
cond = mx.symbol.Activation(data=cond_v, act_type='relu')
body = cos_t
for i in xrange(args.margin/2):
for i in xrange(args.margin//2):
body = body*body
body = body*2-1
new_zy = body*s