553f777ad245ef3caa799151e34e6cc37bbcb11a,gluon/gluoncv2/models/mobilenetv2.py,MobileNetV2,__init__,#MobileNetV2#Any#Any#Any#Any#Any#Any#Any#,93

Before Change

                bn_use_global_stats=bn_use_global_stats,
                activation=ReLU6()))
            in_channels = final_block_channels
            self.features.add(nn.AvgPool2D(
                pool_size=7,
                strides=1))

            self.output = nn.HybridSequential(prefix="")
            self.output.add(conv1x1(
                in_channels=in_channels,
                out_channels=classes,
                use_bias=False))
            self.output.add(nn.Flatten())

    def hybrid_forward(self, F, x):
        x = self.features(x)

After Change

                            strides=strides,
                            bn_use_global_stats=bn_use_global_stats,
                            expansion=expansion,
                            remove_exp_conv=remove_exp_conv))
                        in_channels = out_channels
                self.features.add(stage)
            self.features.add(conv1x1_block(
                in_channels=in_channels,
                out_channels=final_block_channels,
                bn_use_global_stats=bn_use_global_stats,
                activation=ReLU6()))
            in_channels = final_block_channels
            self.features.add(nn.AvgPool2D(
                pool_size=7,
                strides=1))

Instances