Hi,
I’ve got a problem with memory leak during training. I suspect the main cause of that problem is Dataset created by using torchvision.datasets.ImageFolder, (when I used torchvision.datasets.CIFAR10 instead of my dataset the problem does not occur) . I’ve tried to find a solution on similar topics.
Here is my dataset

image.png691×180 5.88 KB

and dataloader:

image.png988×216 9.44 KB

Do you have any idea what could be a reason of a memory leak ?
Thanks in an advance