From 792d1f07ccdbcd9a6a56ae79ac396f10a3c242af Mon Sep 17 00:00:00 2001
From: Gokhan Ozbulak <gokhan.ozbulak@idiap.ch>
Date: Tue, 28 May 2024 15:27:18 +0200
Subject: [PATCH] Fix qa. #69
---
src/mednet/scripts/upload.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py
index ea659e22..f02d0934 100644
--- a/src/mednet/scripts/upload.py
+++ b/src/mednet/scripts/upload.py
@@ -189,7 +189,9 @@ def upload(
click.secho(
f" -> `#accumulations` ({train_data['accumulate-grad-batches']})"
)
- mlflow.log_param("#Accumulations", train_data["accumulate-grad-batches"])
+ mlflow.log_param(
+ "#Accumulations", train_data["accumulate-grad-batches"]
+ )
click.secho(f" -> `epoch (best)` ({best_epoch})")
mlflow.log_param("Epoch (best)", best_epoch)
--
GitLab