From cc6eb213eab0606a4227a648d1ebce58bb943ee9 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Fri, 7 Jun 2024 20:38:14 +0200
Subject: [PATCH] [scripts.experiment] Capture and forward augmentations to
training
---
src/mednet/scripts/experiment.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/mednet/scripts/experiment.py b/src/mednet/scripts/experiment.py
index 8cb233f0..ad4de8bc 100644
--- a/src/mednet/scripts/experiment.py
+++ b/src/mednet/scripts/experiment.py
@@ -49,6 +49,7 @@ def experiment(
parallel,
monitoring_interval,
balance_classes,
+ augmentations,
**_,
): # numpydoc ignore=PR01
r"""Run a complete experiment, from training, to prediction and evaluation.
@@ -89,6 +90,7 @@ def experiment(
parallel=parallel,
monitoring_interval=monitoring_interval,
balance_classes=balance_classes,
+ augmentations=augmentations,
)
train_stop_timestamp = datetime.now()
--
GitLab