Accessing DataModule or DataLoaders within model hooks
Hey, as the title says, I want to access the DataModule or the DataLoader inside the on fit start hook. Is this possible and how can I do it? To be more specifc I want to access my model, when I have access to my DataModule, to get a batch of data, then use it to apply some pruning algorithm on my model.
self.datamodule or self.trainer.train_dataloader in the LightningModule