c0b06f1c3eaaab3d98c14590ff0c8a96dba3ebb1,tests/models/senet.py,CifarSEResNet,forward,#CifarSEResNet#,223

Before Change

        x = self.layer3(x)

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

        return x

After Change

        x = self.layer3(x)

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

        return x

Instances