About memory leakage

[hsvgbkhgbv]

Hi,

Thank you for providing the repo of such a good work. However, when we run the codebase, we find that there exists memory leakage. Could you help deal with that? I guess this is due to the multiprocessing.