I think I have a temporary solution for now. I have split my dataloader into train and test. In train I don’t send the string tuples I was sending before. I only send the image and it’s associated label during training. It kind of solves the issue temporarily, I’ll have to run big tests again. I’ll post if the issue becomes an issue again.
Cheers.