From 51eab8bf829bb4b1746801306cc9c6e624f53528 Mon Sep 17 00:00:00 2001
From: Krzysztof Rudnicki <krzysztofrudnicki0@gmail.com>
Date: Tue, 16 May 2023 23:37:16 +0200
Subject: [PATCH] feat: print info about epoch accuracy

---
 lab5/code/main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lab5/code/main.py b/lab5/code/main.py
index f286bb38..a612d5e0 100644
--- a/lab5/code/main.py
+++ b/lab5/code/main.py
@@ -105,14 +105,14 @@ def single_train_iteration(
     loss.backward()
     training_parameters['optimizer'].step()
     # Print loss value for every learning step
-    """if (batch_idx + 1) % 100 == 0:
+    if (batch_idx + 1) % 100 == 0:
         print(
             f'''
             Epoch [{epoch+1}/{training_parameters['hyperparameters']["num_epochs"]}],
             Step [{batch_idx+1}/{len(training_parameters['loaders']['train_loader'])}],
             Loss: {loss.item():.4f}
             '''
-        )"""
+        )
     # Append loss value for every learning step
     loss_values.append(loss.item())
     return data, training_parameters['optimizer']