Resolving some issues in the resnte101 mxnet symbol

mahyarnajibi · Apr 10, 2018 · ff9d37e · ff9d37e
1 parent e2575b2
commit ff9d37e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/symbols/faster/resnet_mx_101.py b/symbols/faster/resnet_mx_101.py
@@ -99,14 +99,14 @@ def residual_unit_deform(self, data, num_filter, stride, dim_match, name, bn_mom
         if self.fix_bn:
             bn1 = mx.sym.BatchNorm(data=data, fix_gamma=False, eps=2e-5, use_global_stats=True, name=name + '_bn1')
         else:
-            bn3 = mx.sym.BatchNorm(data=conv2, fix_gamma=False, eps=2e-5, momentum=self.momentum, name=name + '_bn3')
+            bn1 = mx.sym.BatchNorm(data=data, fix_gamma=False, eps=2e-5, momentum=self.momentum, name=name + '_bn1')
         act1 = mx.sym.Activation(data=bn1, act_type='relu', name=name + '_relu1')
         conv1 = mx.sym.Convolution(data=act1, num_filter=int(num_filter*0.25), kernel=(1,1), stride=(1,1), pad=(0,0),
                                       no_bias=True, workspace=workspace, name=name + '_conv1')
         if self.fix_bn:
             bn2 = mx.sym.BatchNorm(data=conv1, fix_gamma=False, eps=2e-5, use_global_stats=True, name=name + '_bn2')
         else:
-            bn3 = mx.sym.BatchNorm(data=conv2, fix_gamma=False, eps=2e-5, momentum=self.momentum, name=name + '_bn3')
+            bn2 = mx.sym.BatchNorm(data=conv1, fix_gamma=False, eps=2e-5, momentum=self.momentum, name=name + '_bn2')
         act2 = mx.sym.Activation(data=bn2, act_type='relu', name=name + '_relu2')
         offset = mx.symbol.Convolution(name=name + '_offset', data=act2,
                                                       num_filter=72, pad=(2, 2), kernel=(3, 3), stride=(1, 1),