search

harlanhong / CVPR2022-DaGAN

Official code for CVPR2022 paper: Depth-Aware Generative Adversarial Network for Talking Head Video Generation
https://harlanhong.github.io/publications/dagan.html
Other
957 stars 125 forks source link

What is the minimum GPU memory requirement? #66

Open Pythonpa opened 1 year ago

Pythonpa commented 1 year ago

HI, thanks for your excellent work.

My question is: what is the minimum GPU memory requirement for processing 256 pixels of video?

And I have a Nvidia RTX 3060Ti,8GB memory, when I running demo.py,It was going to tell me "CUDA is OUT of Memory".........sad.

But, if I‘m running "Thin-Plate-Spline-Motion-Model", which is similar to this project, it will work very well.

Inferencer commented 1 year ago

I have an old GPU (GTX 970) with 4GB and it runs fine my driving vid is 60fps and 4k res, my source image is around 540x540 (im still testing things i'm not endorsing using those sizes), i'm not getting any oom errors