search

csxmli2016 / DFDNet

Blind Face Restoration via Deep Multi-scale Component Dictionaries (ECCV 2020)
915 stars 215 forks source link

I am using RTX 2080Ti 11Gb. I lack only 1Gb. Is there anything in the code that I can reduce so I can set gpu_id = 0? #64

Open xamxixixo opened 3 years ago

xamxixixo commented 3 years ago

The title says it all. Lack only 1Gb so I am running it using CPU which takes more than double of the time. So, can I modify the code somehow to make step 3 of DFDNet accept my GPU?

Step 1 and step 2 run well using GPU, so talking about them is no need.

Thank you.