From 04f73835dc5be283eddc62e8dc31062b8bb9df5d Mon Sep 17 00:00:00 2001
From: Thomas Roddick <somahtr@aol.com>
Date: Tue, 30 Mar 2021 18:31:51 +0100
Subject: [PATCH] Fix bug where validation loss does not decrease

---
 train.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/train.py b/train.py
index 04dd4a2..70d623d 100644
--- 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