From 8ef513d2b266c826fa1f441633b01ef26b89e92b Mon Sep 17 00:00:00 2001 From: Byeongho Heo Date: Tue, 3 Dec 2019 13:08:34 +0900 Subject: [PATCH] SSD fix_gamma disabled --- gluoncv/nn/feature.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gluoncv/nn/feature.py b/gluoncv/nn/feature.py index e23f596a6a..89895e7130 100644 --- a/gluoncv/nn/feature.py +++ b/gluoncv/nn/feature.py @@ -143,13 +143,13 @@ def __init__(self, network, outputs, num_filters, use_1x1_transition=True, y, num_filter=num_trans, kernel=(1, 1), no_bias=use_bn, name='expand_trans_conv{}'.format(i), attr={'__init__': weight_init}) if use_bn: - y = mx.sym.BatchNorm(y, name='expand_trans_bn{}'.format(i)) + y = mx.sym.BatchNorm(y, fix_gamma=False, name='expand_trans_bn{}'.format(i)) y = mx.sym.Activation(y, act_type='relu', name='expand_trans_relu{}'.format(i)) y = mx.sym.Convolution( y, num_filter=f, kernel=(3, 3), pad=(1, 1), stride=(2, 2), no_bias=use_bn, name='expand_conv{}'.format(i), attr={'__init__': weight_init}) if use_bn: - y = mx.sym.BatchNorm(y, name='expand_bn{}'.format(i)) + y = mx.sym.BatchNorm(y, fix_gamma=False, name='expand_bn{}'.format(i)) y = mx.sym.Activation(y, act_type='relu', name='expand_reu{}'.format(i)) outputs.append(y) if global_pool: