From 7a1444e533da97e1e05fab77a4e68a60fd3b89bd Mon Sep 17 00:00:00 2001
From: Jakub Henyk <jakhen@st.amu.edu.pl>
Date: Thu, 11 May 2023 21:53:55 +0200
Subject: [PATCH] OMEGA FIX

---
 my_runs/7/run.json | 2 +-
 sacred_train.py    | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/my_runs/7/run.json b/my_runs/7/run.json
index 472b4b0..da02255 100644
--- a/my_runs/7/run.json
+++ b/my_runs/7/run.json
@@ -24,7 +24,7 @@
       ]
     ]
   },
-  "heartbeat": "2023-05-11T19:49:33.658443",
+  "heartbeat": "2023-05-11T19:53:53.820775",
   "host": {
     "ENV": {},
     "cpu": "Intel(R) Core(TM) i7-6700K CPU @ 4.00GHz",
diff --git a/sacred_train.py b/sacred_train.py
index e0fd2e5..b0322b2 100644
--- a/sacred_train.py
+++ b/sacred_train.py
@@ -47,9 +47,9 @@ def trainNet(_run, trainloader, criterion, optimizer, net, epochs=20):
             outputs = net(inputs)
           
             loss = criterion(outputs, labels)
-            print(loss.item())
 
-            _run.log_scalar("training.loss", loss)
+            _run.log_scalar("training.loss", loss.item())
+
             loss.backward()
             optimizer.step()