c15754e49ef498644c714c8e9091471b78b1eb2f,catalyst/contrib/models/classification/mobilenet.py,MobileNetV2,forward,#MobileNetV2#Any#,139

Before Change

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

After Change

    def forward(self, x):
        x = self.encoder(x)
        x = self.conv(x)
        x = self.avgpool(x)
        x = x.view(x.size(0), -1)
        x = self.classifier(x)
        return x

Instances