Qiao
Qiao
> I have the same problem, and the program always stops at the same iteration and then reports an error. What should I do? same here@khwengXU
class UNetPlusPlus(nn.Module): def __init__(self, num_classes, input_channels, block, num_blocks, nb_filter,deep_supervision=False): super(UNetPlusPlus, self).__init__() self.relu = nn.ReLU(inplace = True) self.deep_supervision = deep_supervision self.pool = nn.MaxPool2d(2, 2) self.up = nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True) self.down =...
李工,我又测了一遍,已经可以用了。另外,我想问您一下,我看代码的net.py中提供的是**NestFuse**的网络结构,和您的**RFN-Nest**相似但不同,如编码器的channel两者不同。如果想体验RFN-Nest的训练过程,需要我们自己去改一下? 