From 2f3c547b55df3956266a8a1c1da3837c2f631675 Mon Sep 17 00:00:00 2001
From: Daniel Svitan <daniel.svitan.team7274dev@gmail.com>
Date: Mon, 6 Jan 2025 20:12:40 +0100
Subject: [PATCH] :bug: Fixes test loss value

---
 train_nn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_nn.py b/train_nn.py
index 4f23d34..5625a03 100644
--- a/train_nn.py
+++ b/train_nn.py
@@ -119,7 +119,7 @@ for epoch in range(epochs):
             pred = model(X)
             loss = loss_fn(pred, y)
 
-            test_loss = loss.item() * X.size(0)
+            test_loss += loss.item() * X.size(0)
 
     test_loss /= len(test_dataset)
     test_losses.append(test_loss)