cp-vton-plus icon indicating copy to clipboard operation
cp-vton-plus copied to clipboard

Published 20 hours ago • verified publisher icon minar09

Using multiple GPUs

Open hfarhidzadeh opened this issue 4 years ago • 1 comments

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

hfarhidzadeh avatar Jan 04 '21 01:01 hfarhidzadeh

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.

minar09 avatar Jan 10 '21 04:01 minar09