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
352 stars 121 forks source link

train does not start #34

Closed ikenaga530 closed 3 years ago

ikenaga530 commented 3 years ago

I get the following error. I would like to know what the solution is.

Screenshot from 2020-11-02 23-05-55

thaithanhtuan commented 3 years ago

Look like something wrong with your GPU. To run the code you need at least one GPU (NVIDIA). To run the code without GPU, you should modify the code. Any place you see: .cuda() change it to .cpu() Regards.