Using multiple GPUs - 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

352 stars 121 forks source link

Using multiple GPUs #47

Open hfarhidzadeh opened 3 years ago

hfarhidzadeh commented 3 years ago

Hi,

I specified gpus_id = '0,1', but model just run on first device. I checked the code and couldn't find the "DataParallel" to distribute the model to multiple devices. Please let me know if I am missing something

minar09 commented 3 years ago

Hi, @hfarhidzadeh , normally model should be training on multiple gpus if you set the id values correctly, by dividing the data according to the batch_size//no_of_gpus.