Building an input data pipeline using PyTorch is pretty straight forward using Dataset class from torch.utils.
Howe