From 5f9785d4fd8fcc7a99382fde90f5bd52b824489d Mon Sep 17 00:00:00 2001
From: Maciej Czajka <maccza5@st.amu.edu.pl>
Date: Mon, 9 May 2022 22:14:03 +0200
Subject: [PATCH] update dllib-mlflow.py

---
 Biblioteka_DL/dllib-mlflow.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Biblioteka_DL/dllib-mlflow.py b/Biblioteka_DL/dllib-mlflow.py
index dc5043e..2be6024 100644
--- a/Biblioteka_DL/dllib-mlflow.py
+++ b/Biblioteka_DL/dllib-mlflow.py
@@ -289,7 +289,7 @@ def my_main(epochs):
     loss_fn = nn.CrossEntropyLoss()
     # epochs = 1000
     # epochs = epochs
-    alpha = int(sys.argv[1]) if len(sys.argv) > 1 else 20
+    epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20
     mlflow.log_param("epochs", epochs)
     
     def print_(loss):