Hi there,
I think for your usecase you need all images at every training_step, right?
In that case, I would make two different dataloaders, img_loader and text_loader.
Get a batch that consists of all images before training starts and save it in aimg variable.
Then use your text_loader like a normal dataloader, getting a text batch at every step.
Hope this helps!