From b458f37864381d976c4e9cd4fa8ffffeae1a6c44 Mon Sep 17 00:00:00 2001 From: nttstar Date: Fri, 27 Apr 2018 15:35:20 +0800 Subject: [PATCH] tiny --- src/train_age.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/train_age.py b/src/train_age.py index 242ba9c..5f05265 100644 --- a/src/train_age.py +++ b/src/train_age.py @@ -421,8 +421,8 @@ def train_net(args): else: initializer = mx.init.Xavier(rnd_type='uniform', factor_type="in", magnitude=2) _rescale = 1.0/args.ctx_num - #opt = optimizer.SGD(learning_rate=base_lr, momentum=base_mom, wd=base_wd, rescale_grad=_rescale) - opt = optimizer.Nadam(learning_rate=base_lr, wd=base_wd, rescale_grad=_rescale) + opt = optimizer.SGD(learning_rate=base_lr, momentum=base_mom, wd=base_wd, rescale_grad=_rescale) + #opt = optimizer.Nadam(learning_rate=base_lr, wd=base_wd, rescale_grad=_rescale) som = 20 _cb = mx.callback.Speedometer(args.batch_size, som)