Hmmm. Thank you for your reply.
The issue with my project is it’s MRI images resulting in tensors (s,dimensions) where s = number of slices in an MRI sequence and this varies from one patient to another, so I’m not sure how to use a batch size higher than one.
According to the link, one solution to my issue would be getting a better storage unit? Currently using a 1TB SSD.