torchkit.utils.dataset

torchkit.utils.dataset.infinite_dataset(data_loader)[source]

Create an infinite loop over a torch.utils.DataLoader.

Parameters: data_loader (Iterable) – A torch.utils.DataLoader object.
Yields: Iterator – An iterator over the dataloader that repeats ad infinitum.
Return type: Iterator