Using multi-GPU for Training the model

knazeri / edge-connect

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212

http://openaccess.thecvf.com/content_ICCVW_2019/html/AIM/Nazeri_EdgeConnect_Structure_Guided_Image_Inpainting_using_Edge_Prediction_ICCVW_2019_paper.html

Other

2.5k stars 530 forks source link

Using multi-GPU for Training the model #128

Open Hgit007 opened 4 years ago

Hgit007 commented 4 years ago

Hi Kamyar,

Thanks for your excellent work. Now, I encounter one issue about the training with multi-GPU. And I cannot solve it. Could you please tell me how could I fix the code?