given some dataset X, sufficiently small
create TensorDataset : by converting X to torch.tensors, loading into your gpu (.to(your_gpu)), and using the build in class
create an custom loading class , i.e. just a class that iterates through the TensorDataset.
done