Dynamic Dataloader

[leonardcaquot94]

Hi,

I’m using PyTorch Lightning for a model requiring recursive predictions. After each step, some data is fully processed, while the rest needs to be reinserted into the dataloader for further iterations.

How can I dynamically reinsert partially predicted data into the dataloader during training, ideally within Lightning’s ecosystem? Are there recommended hooks or approaches for this?

Thanks in advance!