After some digging on the forums, CPU RAM Usage Keeps Growing as Training One Cycle, RuntimeError: DataLoader worker is killed by signal, Does DataLoader have multithreading memory leak issues? discuss the same issue. I will share an answer on this thread if I find one.