search

MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.26k stars 538 forks source link

'unet_final_features' referenced before assignment #80

Open mukul-B opened 2 years ago

mukul-B commented 2 years ago

Traceback (most recent call last): File "/usr/local/bin/nnUNet_train", line 33, in sys.exit(load_entry_point('nnunet', 'console_scripts', 'nnUNet_train')()) File "/content/UNetPlusPlus/pytorch/nnunet/run/run_training.py", line 136, in main trainer.initialize(not validation_only) File "/content/UNetPlusPlus/pytorch/nnunet/training/network_training/nnUNetPlusPlusTrainerV2.py", line 121, in initialize self.initialize_network() File "/content/UNetPlusPlus/pytorch/nnunet/training/network_training/nnUNetPlusPlusTrainerV2.py", line 158, in initialize_network self.net_num_pool_op_kernel_sizes, self.net_conv_kernel_sizes, False, True, True) File "/content/UNetPlusPlus/pytorch/nnunet/network_architecture/generic_UNetPlusPlus.py", line 344, in init basic_block, transpconv) File "/content/UNetPlusPlus/pytorch/nnunet/network_architecture/generic_UNetPlusPlus.py", line 475, in create_nest return conv_blocks_localization, tu, unet_final_features UnboundLocalError: local variable 'unet_final_features' referenced before assignment

beargolden commented 1 year ago

Traceback (most recent call last): File "/usr/local/bin/nnUNet_train", line 33, in sys.exit(load_entry_point('nnunet', 'console_scripts', 'nnUNet_train')()) File "/content/UNetPlusPlus/pytorch/nnunet/run/run_training.py", line 136, in main trainer.initialize(not validation_only) File "/content/UNetPlusPlus/pytorch/nnunet/training/network_training/nnUNetPlusPlusTrainerV2.py", line 121, in initialize self.initialize_network() File "/content/UNetPlusPlus/pytorch/nnunet/training/network_training/nnUNetPlusPlusTrainerV2.py", line 158, in initialize_network self.net_num_pool_op_kernel_sizes, self.net_conv_kernel_sizes, False, True, True) File "/content/UNetPlusPlus/pytorch/nnunet/network_architecture/generic_UNetPlusPlus.py", line 344, in init basic_block, transpconv) File "/content/UNetPlusPlus/pytorch/nnunet/network_architecture/generic_UNetPlusPlus.py", line 475, in create_nest return conv_blocks_localization, tu, unet_final_features UnboundLocalError: local variable 'unet_final_features' referenced before assignment

Same problem. Have you solved this problem?