A bug about the UpBlock of the UNet

[szhang963]

The class UpBlock is defined as:

class UpBlock(nn.Module):
    """Upssampling followed by ConvBlock"""

    def __init__(self, in_channels1, in_channels2, out_channels, dropout_p,
                 bilinear=True):

, where the default value of bilinear is "True".
However, the real value of bilinear is not fed into the UpBlock class, as follows:

self.up1 = UpBlock(
    self.ft_chns[4], self.ft_chns[3], self.ft_chns[3], dropout_p=0.0)

Therefore, the 'nn.Upsample' is always selected in the UpBlock.
You can check whether the bug exists or not.

[Xiangde Luo]

Hi, Thanks for your suggestion, actually, these codes of networks are borrowed from some famous and open-access implementations. So, I will check it carefully later, thanks again.
Best,
Xiangde.