Hello!have some problems in modifying the loss function, can you help me solve it? I have added an uncertainty_loss based on dc_loss and ce_loss. I have defined a DC_CE_and_uncertainty_loss class in compound_losses.py. And I have modified _build_loss in nnUNetTrainer.py,as shown in the figure below.
In addition, I also modified self.loss.
After the above modifications, I encountered the following error during training：
Hello!have some problems in modifying the loss function, can you help me solve it? I have added an uncertainty_loss based on dc_loss and ce_loss. I have defined a DC_CE_and_uncertainty_loss class in compound_losses.py. And I have modified _build_loss in nnUNetTrainer.py,as shown in the figure below.
In addition, I also modified self.loss.
After the above modifications, I encountered the following error during training：
