Inference with cpu - Githubissues

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

Closed dnalexen closed 2 years ago

dnalexen commented 3 years ago

I tried to disable the cuda, but the file test.py didn't run. Is there a way to run an inference with a cpu?

thaithanhtuan commented 3 years ago

How do you disable the cuda? You can change the code from : .cuda() to .cpu()

dnalexen commented 3 years ago

@thaithanhtuan Thank you.