MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.32k stars 542 forks source link

AttributeError: 'Generic_UNetPlusPlus' object has no attribute 'upsample_mode' #89

Open SalamanderXing opened 2 years ago

SalamanderXing commented 2 years ago

Line 458 in the file: https://github.com/MrGiovanni/UNetPlusPlus/blob/master/pytorch/nnunet/network_architecture/generic_UNetPlusPlus.py references self.upsample_mode, but this is not defined. To fix it it was necessary to replace upsample_mode = ... to self.upsample_mode in init.py of class Generic_UNetPlusPlus.