From ce314d56318384fc979ead0623b744c1a13cc659 Mon Sep 17 00:00:00 2001
From: Gokhan Ozbulak <gokhan.ozbulak@idiap.ch>
Date: Sat, 25 May 2024 00:20:22 +0200
Subject: [PATCH] Fix qa #69.
---
src/mednet/scripts/upload.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py
index 1bd1add1..e3a4ff75 100644
--- a/src/mednet/scripts/upload.py
+++ b/src/mednet/scripts/upload.py
@@ -157,7 +157,7 @@ def upload(
evaluation_data = evaluation_data["test"]
# get lowest validation epoch
- best_epoch = str(train_model_file).split('.')[0].split('=')[1]
+ best_epoch = str(train_model_file).split(".")[0].split("=")[1]
experiment_name = (
experiment_name
@@ -185,9 +185,11 @@ def upload(
]:
click.secho(f" -> `{k}` ({train_data[k]})")
mlflow.log_param(k, train_data[k])
-
- click.secho(f" -> `#accumulations` ({train_data['batch-chunk-count']})")
- mlflow.log_param("#Accumulations", train_data['batch-chunk-count'])
+
+ click.secho(
+ f" -> `#accumulations` ({train_data['batch-chunk-count']})"
+ )
+ mlflow.log_param("#Accumulations", train_data["batch-chunk-count"])
click.secho(f" -> `epoch (best)` ({best_epoch})")
mlflow.log_param("Epoch (best)", best_epoch)
--
GitLab