'unet_final_features' referenced before assignment

[mukul-B]

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]

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?