From 3d7704bb27b4c6cf0943ae3d638c80da5625b1a9 Mon Sep 17 00:00:00 2001
From: Jakub Henyk <jakhen@st.amu.edu.pl>
Date: Thu, 11 May 2023 21:32:22 +0200
Subject: [PATCH] fix?

---
 sacred_train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/sacred_train.py b/sacred_train.py
index 0623a99..0488419 100644
--- a/sacred_train.py
+++ b/sacred_train.py
@@ -63,7 +63,7 @@ def my_config():
     epochs = 10
     learning_rate = 0.001
 
-@ex.main
+@ex.automain
 def my_main(epochs, learning_rate, _run):
 
     ex.open_resource("Customers.csv", "r")
@@ -97,7 +97,7 @@ def my_main(epochs, learning_rate, _run):
 
     ex.add_artifact("cifar_net.pth")
 
-if __name__ == '__main__':
+#if __name__ == '__main__':
 
-    ex.run()
+    #ex.run()