update dllib-mlflow.py

2022-05-09 22:14:03 +02:00 · 2022-05-09 22:14:03 +02:00 · 5f9785d4fd
commit 5f9785d4fd
parent 8f297e82de
1 changed files with 1 additions and 1 deletions
--- 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):