- Jul 26, 2023
-
-
André Anjos authored
[data.datamodule] Double-check model_transforms are set before datasets are instantiated; Remove model_transforms from constructor (non-sensical); Improve documentation on model-transforms
-
André Anjos authored
-
André Anjos authored
-
-
André Anjos authored
-
André Anjos authored
[ptbench.data.datamodule] Implemented typing, added more logging, implemented sampler-balancing depending on sample class prevalence, parallelized dataset caching
-
Implemented BCEWithLogitsLoss reweighting function Removed save_hyperparameters from model Apply augmentation transforms on singular images Fixed model summary
-
-
André Anjos authored
-
-
André Anjos authored
-
André Anjos authored
[datamodule] Slightly streamlines the datamodule approach; adds documentation; adds type annotations; adds TODOs
-
Created base class for every LightningDataModule, that can be inherited from. For each protocol, the user should create a new class inheriting from BaseDataModule and implement the setup() function, in which the dataset is defined.
-
-
DataModule is generic and not tied to a specific dataset.
-
-