Running dataloader concurrently in CPU with the model (in GPU) using pytorch lightning #17619
Unanswered
jmoork
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 1 comment
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Is it possible to run the dataloader in CPU concurrently with the GPU running the model forward processing in pytorch lightning? This essentially allows the CPU dataloader to fetch the next batch of data while GPU is processing the current batch.
Beta Was this translation helpful? Give feedback.
All reactions