From f8a39f8f148fff51cdfcf967cb19d70ab3e80e7b Mon Sep 17 00:00:00 2001 From: nttstar Date: Mon, 29 Jan 2018 23:03:01 +0800 Subject: [PATCH] add FGHIJ end setting --- src/symbols/symbol_utils.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/src/symbols/symbol_utils.py b/src/symbols/symbol_utils.py index c4493dc..70accd9 100644 --- a/src/symbols/symbol_utils.py +++ b/src/symbols/symbol_utils.py @@ -23,13 +23,19 @@ def get_fc1(last_conv, num_classes, fc_type): fc1 = mx.sym.FullyConnected(data=body, num_hidden=num_classes, name='pre_fc1') fc1 = mx.sym.BatchNorm(data=fc1, fix_gamma=True, eps=2e-5, momentum=bn_mom, name='fc1') elif fc_type=='F': - bn1 = mx.sym.BatchNorm(data=body, fix_gamma=False, eps=2e-5, momentum=bn_mom, name='bn1') - relu1 = Act(data=bn1, act_type='relu', name='relu1') - body = mx.symbol.Dropout(data=relu1, p=0.4) + body = mx.sym.BatchNorm(data=body, fix_gamma=False, eps=2e-5, momentum=bn_mom, name='bn1') + body = mx.symbol.Dropout(data=body, p=0.4) + fc1 = mx.sym.FullyConnected(data=body, num_hidden=num_classes, name='fc1') + elif fc_type=='G': + body = mx.sym.BatchNorm(data=body, fix_gamma=False, eps=2e-5, momentum=bn_mom, name='bn1') + fc1 = mx.sym.FullyConnected(data=body, num_hidden=num_classes, name='fc1') + elif fc_type=='H': + fc1 = mx.sym.FullyConnected(data=body, num_hidden=num_classes, name='fc1') + elif fc_type=='I': + body = mx.sym.BatchNorm(data=body, fix_gamma=False, eps=2e-5, momentum=bn_mom, name='bn1') fc1 = mx.sym.FullyConnected(data=body, num_hidden=num_classes, name='pre_fc1') fc1 = mx.sym.BatchNorm(data=fc1, fix_gamma=True, eps=2e-5, momentum=bn_mom, name='fc1') - elif fc_type=='G': - body = mx.symbol.Dropout(data=body, p=0.4) + elif fc_type=='J': fc1 = mx.sym.FullyConnected(data=body, num_hidden=num_classes, name='pre_fc1') fc1 = mx.sym.BatchNorm(data=fc1, fix_gamma=True, eps=2e-5, momentum=bn_mom, name='fc1') else: