4ee8ae757fd0701f8030f9fc46d6130b6d289123,models/unet/unet_model.py,UnetModel,__init__,#UnetModel#Any#Any#Any#Any#Any#,68

Before Change

        self.up_transpose_conv += [nn.ConvTranspose2d(ch, ch, kernel_size=2, stride=2, bias=False)]
        self.up_conv += [ConvBlock(ch * 2, ch, drop_prob)]

        self.conv2 = nn.Sequential(
            nn.Conv2d(ch, ch // 2, kernel_size=1),
            nn.Conv2d(ch // 2, out_chans, kernel_size=1),
            nn.Conv2d(out_chans, out_chans, kernel_size=1),
        )

    def forward(self, input):
        
        Args:

After Change

            self.up_conv += [ConvBlock(ch * 2, ch, drop_prob)]
            ch //= 2

        self.up_transpose_conv += [TransposeConvBlock(ch * 2, ch)]
        self.up_conv += [
            nn.Sequential(
                ConvBlock(ch * 2, ch, drop_prob),
                nn.Conv2d(ch, self.out_chans, kernel_size=1, stride=1),

Instances