How to call the data loader inside pl.LightningModule? #8973
Unanswered
KinWaiCheuk
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
My scheduler takes the total number of steps
len(train_loader)*epochs
as one of its arguments.I know inside the pl.LightningModule, I can get the number of epochs via
self.trainer.max_epochs
.But how to get the data_loader?
I have tried
self.trainer.train_dataloaders
,self.train_dataloaders
, but it does not return me the data loader I want.Beta Was this translation helpful? Give feedback.
All reactions