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

tiny

Browse Source
This commit is contained in:
Jia Guo
2018-01-03 23:28:14 +08:00
parent 925714d961
commit 77f7f0a856
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

@@ -396,7 +396,7 @@ def train_net(args):
if args.loss_type==1 and args.num_classes>40000:
if (args.loss_type==1 or args.loss_type==3) and args.num_classes>40000:
args.beta_freeze = 5000
args.gamma = 0.06
@@ -595,7 +595,7 @@ def train_net(args):
save_step = [0]
if len(args.lr_steps)==0:
lr_steps = [40000, 60000, 80000]
if args.loss_type==1:
if args.loss_type==1 or args.loss_type==3:
lr_steps = [100000, 140000, 160000]
p = 512.0/args.batch_size
for l in xrange(len(lr_steps)):