search

MRzzm / DINet

The source code of "DINet: deformation inpainting network for realistic face visually dubbing on high resolution video."
963 stars 171 forks source link

12G GPU train Error?? #71

Open icemanyandy opened 11 months ago

icemanyandy commented 11 months ago

python train_DINet_clip.py XXXXXX

oh no....

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 12.00 GiB total capacity; 11.23 GiB already allocated; 0 bytes free; 11.29 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

any ideas？

NaMoCv commented 10 months ago

降低分辨率 or 换更大显存的卡，dinet部分建议24G显存以上。

A11enCheung commented 4 months ago

same problem, any ideas to solve it？