Fix bug where validation loss does not decrease

tom-roddick · Mar 30, 2021 · 04f7383 · 04f7383
1 parent 38886ea
commit 04f7383
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -158,7 +158,6 @@ def compute_loss(pred_encoded, gt_encoded, loss_weights=[1., 1., 1., 1.]):
         'total' : float(total_loss) 
     }
 
-    total_loss = score_loss
     return total_loss, loss_dict