YuvalNirkin / fsgan

FSGAN - Official PyTorch Implementation
https://nirkin.com/fsgan
Creative Commons Zero v1.0 Universal
749 stars 147 forks source link

Is there a way to run it on CPU? #134

Closed ghost closed 3 years ago

ghost commented 3 years ago

I wanted to try Face reenactment, but I only have a 2GB GPU, and I'm not sure how to run it on Google Colab. I've tried reducing batch size, but it doesn't help. So I was wondering if there is some way to run it on CPU?

P.S. I was able to run it on Google Colab. I assume it's not possible to easily run CUDA code on CPU (if I understand correctly that it uses CUDA), so I'm closing this issue.