I have tested my code and it trains fine when using a smaller subset of my data (40 items). But when I move to the full dataset it just gets stuck loading the data. I am using a batch size of 25 and the larger dataset has ~300 items. Each item in the dataset consists of 7 224x224 images.