Rerun your code with num_workers=0 and check, if an entire epoch can be loaded properly.
If that’s the case then the usage of multiprocessing might be causing some issues, e.g. if not enough shared memory is available etc.