Skip to content
Snippets Groups Projects
Commit efe22d31 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

[datamodule] Use train split if no validation split available

parent 2a85e330
No related branches found
No related tags found
1 merge request!46Create common library
......@@ -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-")
]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment