Closed foxyear-kyumin closed 8 months ago
I am run my tests on T4 and k8 gpu (16 and 8 GB memory).
This errors depends on your GPU and it is mainly happening since the audio file you are testing might be long enough and hence the torch model is running OOM. The longer the audio file, the more gpu memory utilization is required. To fix this error, I have modified processing the audio file into smaller chunks of 10 secs. This would prevent running into OOM on your GPU. Please pull the latest commits and test again.
Let me know if you still have any other problems.
Traceback (most recent call last): File "D:\APP\DINet_optimized-main\train_DINet_clip.py", line 198, in
Good job!Thanks Elsaam2y. I want know how to train syncnet.pt,can give me any suggestion?
For the syncnet I tried to get some inspirations and follow the syncnet implementation for wav2lip. However the final results were not always consistent and thats why I thought about using the original trained DINet model but with a trained and lightweight audio mapping model.
rch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 684.00 MiB (GPU 0; 24.00 GiB total capacity; 20.99 GiB already allocated; 0 bytes free; 21.17 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF