update dllib-mlflow.py

2022-05-11 15:25:58 +02:00 · 2022-05-11 15:25:58 +02:00 · f8af56e2ca
commit f8af56e2ca
parent 777e10f0ef
1 changed files with 1 additions and 1 deletions
--- a/Biblioteka_DL/dllib-mlflow.py
+++ b/Biblioteka_DL/dllib-mlflow.py
@ -310,7 +310,7 @@ def my_main(epochs):
        loss.backward()  # Gradients
        optimizer.step()  # Update

-    mlflow.log_param("loss", loss.item)
+    mlflow.log_param("loss", loss.item())

    # Prediction
    x_test = Variable(torch.from_numpy(features_test_g)).float()