c45c30b6a0efd72f8bd3d3de67655dfe58bcfc4e,examples/model_compress/models/mobilenet_v2.py,MobileNetV2,forward,#MobileNetV2#Any#,109

Before Change

    def forward(self, x):
        x = self.features(x)
        x = x.mean().mean(2)
        x = self.classifier(x)
        return x

After Change

        // it"s same with .mean(3).mean(2), but
        // speedup only suport the mean option
        // whose output only have two dimensions
        x = x.mean([2, 3])
        x = self.classifier(x)
        return x

Instances