search

svip-lab / impersonator

PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis
https://svip-lab.github.io/project/impersonator
Other
1.73k stars 318 forks source link

I run it on win10 successfully, but it always use cpu not GPU #66

Open Hewentao123 opened 3 years ago

Hewentao123 commented 3 years ago

On my virtual environment , I used the code to test CUDA-GPU successfully and I set the gpu-ids=0, but it still use cpu. I don't know the reason. could you help me?

piaozhx commented 3 years ago

that's impossible because we do not have cpu version. if you run the code successfully, it will definitely use gpu.

Hewentao123 commented 3 years ago

Thanks for your revision. Maybe, I need to set the availiable num_workers and increase the batsize as the GPU has been in idle waiting state. Thanks！