In your screenshot, it seems like there is no leak, just high usage.
If you use many workers or if your dataset has larger images or larger batchsize, then the memory needed to load the data will be bigger.