c8fadd2d970f1c62ae8a842464056263f8d1232f,resnet.py,myResnet,forward,#myResnet#Any#,31

Before Change

        x = self.resnet.layer3(x)
        x = self.resnet.layer4(x)

        fc = x.mean().mean().squeeze()
        att = spatialAdaAvgPool(x,14,14).squeeze().permute(1, 2, 0)

        return fc, att

After Change

        x = self.layer3(x)
        x = self.layer4(x)

        x = self.avgpool(x)
        x = x.view(x.size(0), -1)
        x = self.fc(x)

        return x

Instances