sangyun884 / HR-VITON

Official PyTorch implementation for the paper High-Resolution Virtual Try-On with Misalignment and Occlusion-Handled Conditions (ECCV 2022).
825 stars 171 forks source link

Why am I driving multiple GPUS for training still slow? #60

Open 111wukong opened 1 year ago

111wukong commented 1 year ago

Does this project not support multiple Gpus.Why am I driving multiple Gpus for training and still slow.

hyyuan123 commented 1 year ago

I have encountered the same problem. Have you resolved it

meetncl commented 9 months ago

Hey @111wukong, I am trouble while training the model, Can you share the script for the same??

111wukong commented 9 months ago

Hey @111wukong, I am trouble while training the model, Can you share the script for the same??

It should be possible to address this through model parallelism. You can give it a try.