Access dataloader on a specific GPU instance

[vobecant]

Hi,

During the training, I need to access the dataloader sampled for the specific device. However, when I call self.train_dataloader(), I get the training dataloader for all the devices, i.e., the non-sampled one. Can you please help me and tell me how to access the device-specific dataloader?

Thanks a lot in advance.