Travis Zhang

Results 3 comments of Travis Zhang

Hi, I've met the same problem. The job is done but the memory is still in use, and eventually cause OOM and CUDA out of memory.

I've tried the unload_all_models() function, but it can only free part of the memory.