Thanks for your efforts to maintain the toolkit, guys. May I ask for some help about how to deal with multiple dataloaders in LightningModule? #19908
Unanswered
onbigion13
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
-
I'm writing a project where I need test the model on multiple test datasets. So I return a dict consisting of multiple dataloaders in test_dataloader() method in LightningDataModule. Now I suffer from how to handle the computation, logging and saving behaviours for each testloader in LightningModule. I alreay know that **_step() method could compute for each loader with dataloder_idx. But how about other hooks in LightningModules?
I really appreciate your help, if you could spare some help on me.
Beta Was this translation helpful? Give feedback.
All reactions