From 50a2e2d2f098b8b26d8b9bab1aa95033eaa44e68 Mon Sep 17 00:00:00 2001 From: nttstar Date: Thu, 25 Oct 2018 19:35:17 +0800 Subject: [PATCH] tiny on network symbols --- src/symbols/fmobilefacenet.py | 2 +- src/symbols/fresnet.py | 5 +---- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/src/symbols/fmobilefacenet.py b/src/symbols/fmobilefacenet.py index 1b68587..ee31fc9 100644 --- a/src/symbols/fmobilefacenet.py +++ b/src/symbols/fmobilefacenet.py @@ -47,7 +47,7 @@ def get_symbol(num_classes, **kwargs): bn_mom = kwargs.get('bn_mom', 0.9) wd_mult = kwargs.get('wd_mult', 1.) version_output = kwargs.get('version_output', 'GNAP') - assert version_output=='GDC' or version_output=='GNAP' + #assert version_output=='GDC' or version_output=='GNAP' fc_type = version_output data = mx.symbol.Variable(name="data") data = data-127.5 diff --git a/src/symbols/fresnet.py b/src/symbols/fresnet.py index 01582dd..6c97419 100644 --- a/src/symbols/fresnet.py +++ b/src/symbols/fresnet.py @@ -479,10 +479,7 @@ def residual_unit(data, num_filter, stride, dim_match, name, bottle_neck, **kwar elif uv==4: return residual_unit_v4(data, num_filter, stride, dim_match, name, bottle_neck, **kwargs) else: - if version_input<=1: - return residual_unit_v3(data, num_filter, stride, dim_match, name, bottle_neck, **kwargs) - else: - return residual_unit_v3_x(data, num_filter, stride, dim_match, name, bottle_neck, **kwargs) + return residual_unit_v3(data, num_filter, stride, dim_match, name, bottle_neck, **kwargs) def resnet(units, num_stages, filter_list, num_classes, bottle_neck, **kwargs): bn_mom = kwargs.get('bn_mom', 0.9)