lucidrains / lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two
MIT License
1.62k stars 220 forks source link

Multi GPU utilization problem. #121

Open JohnnyRacer opened 2 years ago

JohnnyRacer commented 2 years ago

Hi, for multi GPU training my GPU0 is using significantly more VRAM and power than GPU1, is this normal or is there someway to balance the load?

image