How to reset train data loader every epoch end?? #17158
Unanswered
elephantoid
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 1 comment
-
There is the Trainer Flag https://lightning.ai/docs/pytorch/stable/common/trainer.html#reload-dataloaders-every-n-epochs) |
Beta Was this translation helpful? Give feedback.
0 replies
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.
-
currently, I updated PL from 1.6.5 to 2.0.0.
in past version, I use trainer.reset_train_dataloader() to reset and resampling dataloader every epoch.
why i've used that function is I want to reset only train loader in fit stage.
so I made a custom callback and run when on_train_epoch_end
but updated version, reset_train_dataloader is deprecated.
how can i do or approach like reset_train_dataloader behavior?
Beta Was this translation helpful? Give feedback.
All reactions