8cc023c568f2cdcdf5c34c5fa1b76090073bca73,gluon/gluoncv2/models/resattnet.py,ResAttNet,forward,#ResAttNet#Any#,342

Before Change

    def forward(self, x):
        x = self.features(x)
        x = x.view(x.size(0), -1)
        x = self.output(x)
        return x

After Change

            self.pool = nn.MaxPool2D(
                pool_size=3,
                strides=2,
                padding=1)
            self.res_blocks = ResBlockSequence(
                in_channels=in_channels,
                out_channels=out_channels,

Instances