From a8c1a1ba36147cd8f23961c5a211cca132d47756 Mon Sep 17 00:00:00 2001
From: Gokhan Ozbulak <gokhan.ozbulak@idiap.ch>
Date: Tue, 28 May 2024 15:25:32 +0200
Subject: [PATCH] Update reference to batch accumulation flag. #69
---
src/mednet/scripts/upload.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py
index e3a4ff75..ea659e22 100644
--- a/src/mednet/scripts/upload.py
+++ b/src/mednet/scripts/upload.py
@@ -187,9 +187,9 @@ def upload(
mlflow.log_param(k, train_data[k])
click.secho(
- f" -> `#accumulations` ({train_data['batch-chunk-count']})"
+ f" -> `#accumulations` ({train_data['accumulate-grad-batches']})"
)
- mlflow.log_param("#Accumulations", train_data["batch-chunk-count"])
+ mlflow.log_param("#Accumulations", train_data["accumulate-grad-batches"])
click.secho(f" -> `epoch (best)` ({best_epoch})")
mlflow.log_param("Epoch (best)", best_epoch)
--
GitLab