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?

[xamxixixo]

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.