Memory leak during the forward pass in eval.py.
quenot opened this issue · 0 comments
quenot commented
Thanks for sharing this great work.
I found that there is a GPU memory leak during the forward pass in eval.py (line 34).
This prevents to complete the evaluation as it crashes at about 7700 samples on a 24GB GPU.
Best regards,
Georges.