No worries. I have not done any work with zarr before, and I am not planning to since the speed is not an issue anymore at the moment. GPU utilization stays the same over the whole training. Weirdly, I had some issues before, depending on the pytorch version, as mentioned above. But using pin_memory=True seemed to solve it.