Ya you are right I missed that detail. If it is the case that the image file is intact, it would imply that in the second epoch the dataloader isnt reading from the actual file. I might not make sense but so seems the problem atm, i am curious at this point,
is the pin_memory true, then try false?
I cant think of any other way you are not accessing ‘fresh’ data