Dropping apply_transforms from LightningModule and LightningDataModule.
Adding the transforms argument, which accepts a list of transformation objects which can be specified in yaml.
For LightningModule, the transformation objects should inherit augment.cuda.CudaBase, while for LightningDataModule it's augment.cpu.CPUBase.
Dropping
apply_transformsfromLightningModuleandLightningDataModule. Adding thetransformsargument, which accepts a list of transformation objects which can be specified in yaml. ForLightningModule, the transformation objects should inheritaugment.cuda.CudaBase, while forLightningDataModuleit'saugment.cpu.CPUBase.close #22