[GitHub] ppwwyyxx commented on a change in pull request #8414: Fix the strides in Gluon version of ResNet v1

[GitBox <gi...@apache.org>]

ppwwyyxx commented on a change in pull request #8414: Fix the strides in Gluon version of ResNet v1
URL: https://github.com/apache/incubator-mxnet/pull/8414#discussion_r187731434
 
 

 ##########
 File path: python/mxnet/gluon/model_zoo/vision/resnet.py
 ##########
 @@ -102,10 +102,10 @@ class BottleneckV1(HybridBlock):
     def __init__(self, channels, stride, downsample=False, in_channels=0, **kwargs):
         super(BottleneckV1, self).__init__(**kwargs)
         self.body = nn.HybridSequential(prefix='')
-        self.body.add(nn.Conv2D(channels//4, kernel_size=1, strides=1))
+        self.body.add(nn.Conv2D(channels//4, kernel_size=1, strides=stride))
 
 Review comment:
   In my implementation, using stride=2 on 1x1 conv has about 0.3 lower top-1 accuracy.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services