Hi, this is a very great framework. I was wondering in torch.utils.data.Dataset we can edit the getitem method in the custom dataset. I could not find anything in LightningDataModule. Is there any way to do that?
LightningDataModule just holds your dataloaders. You still need to use native PyTorch Dataset/Dataloaders to work with data.
you can read more about it here: https://pytorch-lightning.readthedocs.io/en/latest/datamodules.html