”Remind“——Using multi GPU training will get stuck.

Lotayou / Face-Renovation

Official repository of the paper "HiFaceGAN: Face Renovation via Collaborative Suppression and Replenishment".

https://arxiv.org/abs/2005.05005

GNU General Public License v3.0

284 stars 49 forks source link

”Remind“——Using multi GPU training will get stuck. #37

Closed LeeHW-THU closed 3 years ago

LeeHW-THU commented 3 years ago

I get stuck when I train with multi GPU. Using single card can run normally.

Lotayou commented 3 years ago

Thanks for noting, when I did this project I found a single GPU is adequate for most cases so I didn't pay close attention to multi-GPU support. This project is largely inherited from SPADE，maybe you can look into it for a possible solution.