maflx

Results 2 comments of maflx

I had the same problem. In my case it was caused because I was working in a virtual environment and when I compiled the modules they were created for the...

I have the same problem. The cleanup() doesn't clean the memory: https://github.com/ostris/ai-toolkit/blob/58f9d01c2bd7edfb5de0ff61dd564481705cdb89/toolkit/job.py#L44 I have added: `del job.process` `del job` `gc.collect()` `torch.cuda.empty_cache()` reducing the memory but I still have some memory...