Update nll_co_teaching.py

HiLab-git · Oct 4, 2024 · 908eb6f · 908eb6f
1 parent bbc75bf
commit 908eb6f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pymic/net_run/noisy_label/nll_co_teaching.py b/pymic/net_run/noisy_label/nll_co_teaching.py
@@ -50,7 +50,8 @@ def training(self):
         rampup_end   = nll_cfg.get('rampup_end', iter_max)
 
         train_loss_no_select1, train_loss_no_select2  = 0, 0
-        train_loss1, train_avg_loss2 = 0, 0
+        train_loss1, train_avg_loss1 = 0, 0
+        train_loss2, train_avg_loss2 = 0, 0
         train_dice_list = []
         data_time, gpu_time, loss_time, back_time = 0, 0, 0, 0
         self.net.train()