update dllib-mlflow.py

2022-05-09 22:25:23 +02:00 · 2022-05-09 22:25:23 +02:00 · c2a7fbbf25
commit c2a7fbbf25
parent 5f9785d4fd
1 changed files with 21 additions and 21 deletions
--- a/Biblioteka_DL/dllib-mlflow.py
+++ b/Biblioteka_DL/dllib-mlflow.py
@ -296,7 +296,7 @@ def my_main(epochs):
        print ("The loss calculated: ", loss)


-    # Not using dataloader
+    with mlflow.start_run() as run:
    	x_train, y_train = Variable(torch.from_numpy(features_train_g)).float(), Variable(torch.from_numpy(labels_train_g)).long()
    	for epoch in range(1, epochs + 1):
        	print("Epoch #", epoch)