justheuristic/prefetch_generator

memory leak when using prefetch generator in DDP

Opened this issue · 0 comments

After using the prefetch generator, the dataloader is not automatically released after each epoch under multi-gpu-DDP, resulting in a memory leak, has this been encountered by anyone?