Making use of LightningDataModule and simplification of data loading
Compare changes
Files
7- André Anjos authored
[ptbench.data.datamodule] Implemented typing, added more logging, implemented sampler-balancing depending on sample class prevalence, parallelized dataset caching