Move dataset splitting logic to the LightningDataModule
#50
Labels
refactor
Refers to various code refactoring
LightningDataModule
#50
This would result in a very clean training script, and have a dedicated function for dataset splitting.
The text was updated successfully, but these errors were encountered: