search

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
346 stars 120 forks source link

CPU version #21

Closed Bouncer51 closed 4 years ago

Bouncer51 commented 4 years ago

Does this model work on cpu version while testing/inference. if yes, please tell me changes I would have to make.

Thanks

minar09 commented 4 years ago

Hi @Bouncer51 , were you able to run in CPU? It would be helpful for others if you provide some details on how you did it. Thanks.

Bouncer51 commented 4 years ago

yea, i was able to run on cpu version, wherever there was .cuda() in code i removed it and i commented lines model.cuda(). it worked fine for testing 1 image on cpu results were same as gpu version.