minar09 / cp-vton-plus

Official implementation for "CP-VTON+: Clothing Shape and Texture Preserving Image-Based Virtual Try-On", CVPRW 2020
https://minar09.github.io/cpvtonplus/
MIT License
343 stars 120 forks source link

GPU for inference #88

Closed AnjaliSinghTW closed 2 years ago

AnjaliSinghTW commented 2 years ago

Is GPU mandatory for inference? If not, can you provide us instructions to do inference on CPU?

vinodbukya6 commented 2 years ago

Hi, you can check GitHub repo: https://github.com/vinodbukya6/cp-VTryon-plus-Flask-App

or you can replace .cuda() with .cpu() in test.py, networks.py

AnjaliSinghTW commented 2 years ago

Hi, Thank you for your reply. I will do as you suggested. Regards

On Fri, Apr 8, 2022 at 8:38 PM BUKYA VINOD @.***> wrote:

Hi, you can check my GitHub repo: https://github.com/vinodbukya6/cp-VTryon-plus-Flask-App

or you can replace .cuda() with .cpu() in test.py, networks.py

— Reply to this email directly, view it on GitHub https://github.com/minar09/cp-vton-plus/issues/88#issuecomment-1092971136, or unsubscribe https://github.com/notifications/unsubscribe-auth/AYFTBZCZBRRPFPINSRXPLELVEBDWJANCNFSM5SVL66IA . You are receiving this because you authored the thread.Message ID: @.***>