def07745ebcbe08ebb2fbba124bd07873edc8c9c,models/unet.py,DecoderBlockV2,__init__,#DecoderBlockV2#Any#Any#Any#Any#,45

Before Change

                following link https://distill.pub/2016/deconv-checkerboard/
            

            self.block = nn.Sequential(
                ConvRelu(in_channels, middle_channels),
                nn.ConvTranspose2d(
                    middle_channels,
                    out_channels,
                    kernel_size=4,
                    stride=2,
                    padding=1),
                nn.ReLU(inplace=True)
            )
        else:
            self.block = nn.Sequential(
                nn.Upsample(scale_factor=2, mode="bilinear"),

After Change

            in_channels=3,
            num_filters=64,
            num_blocks=4):
        super().__init__()

        self.encoder = Encoder(in_channels, num_filters, num_blocks)
        self.decoder = Decoder(num_filters, num_blocks - 1)
        self.final = nn.Conv2d(num_filters, num_classes, 1)

Instances