Pytorch dataloader restart connection with remote data lake possibility

Hi Pytorch Team

We are using dataloader to load some data from Azure data lake, we will put the Azure data lake client in data loader init function, however, there is risk that connection may be closed or timeout, etc. so I wonder is there a way for dataloader to auto restart , and more importantly, resume from last loaded index and item? what is best practice here?