8a0becb6d6d3271458faa01d0989c7a83a20b88c,model.py,Generator,__init__,#Generator#Any#,6

Before Change

        self.conv1 = nn.Conv2d(3, 32, (5, 5), (1, 1), (2, 2))
        self.conv2 = nn.Conv2d(32, 64, (3, 3), (1, 1), (1, 1))
        self.conv3 = nn.Conv2d(64, 64, (3, 3), (1, 1), (1, 1))
        self.conv4 = nn.Conv2d(64, 32, (3, 3), (1, 1), (1, 1))
        self.conv5 = nn.Conv2d(32, 3 * (upscale_factor ** 2), (3, 3), (1, 1), (1, 1))
        self.prelu = nn.PReLU()
        self.pixel_shuffle = nn.PixelShuffle(upscale_factor)

After Change

        self.conv6 = nn.Conv2d(in_channels=256, out_channels=128, kernel_size=3, stride=1, padding=1, groups=32)
        self.conv7 = nn.Conv2d(in_channels=128, out_channels=3 * (upscale_factor ** 2), kernel_size=5, stride=1,
                               padding=2)
        self.lrelu = nn.LeakyReLU(0.2, inplace=True)
        self.pixel_shuffle = nn.PixelShuffle(upscale_factor)

    def forward(self, x):
        x = self.lrelu(self.conv1(x))

Instances