From efe22d31395f683476eaed5b7ee838c06efac33f Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Fri, 19 Apr 2024 14:35:54 +0200
Subject: [PATCH] [datamodule] Use train split if no validation split available
---
src/mednet/libs/common/data/datamodule.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/src/mednet/libs/common/data/datamodule.py b/src/mednet/libs/common/data/datamodule.py
index e6d0ca0e..3bfcb9f5 100644
--- a/src/mednet/libs/common/data/datamodule.py
+++ b/src/mednet/libs/common/data/datamodule.py
@@ -570,7 +570,14 @@ class ConcatDataModule(lightning.LightningDataModule):
The list of validation dataset names.
"""
- return ["validation"] + [
+ validation_split_name = "validation"
+ if "validation" not in self.splits.keys():
+ logger.warning(
+ "No split named 'validation', the training split will be used for validation instead."
+ )
+ validation_split_name = "train"
+
+ return [validation_split_name] + [
k for k in self.splits.keys() if k.startswith("monitor-")
]
--
GitLab