The best way for you is to create a Dataset class 36 for your specific data, which will enable you to use PyTorch Dataloader. Furthermore, you can use the tremendous amount of neat tutorials to train a/the model and tests it your data.