From 77f7f0a85670c9dd4a3d8e559bb093d69b5ce3ca Mon Sep 17 00:00:00 2001 From: Jia Guo Date: Wed, 3 Jan 2018 23:28:14 +0800 Subject: [PATCH] tiny --- src/train_softmax.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/train_softmax.py b/src/train_softmax.py index 513fdc9..568a278 100644 --- a/src/train_softmax.py +++ b/src/train_softmax.py @@ -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)):