How many GPUs does HistoGAN need for training？

mahmoudnafifi / HistoGAN

Reference code for the paper HistoGAN: Controlling Colors of GAN-Generated and Real Images via Color Histograms (CVPR 2021).

MIT License

267 stars 27 forks source link

How many GPUs does HistoGAN need for training？ #10

Closed Radium98 closed 2 years ago

Radium98 commented 2 years ago

Can you tell me your training configuration？Such how many GPUs you need for training and how long it takes for training?

mahmoudnafifi commented 2 years ago

All models used in the paper were trained using --network_capacity 16 due to the hardware limitation. You can try increasing this number for a better quality if your dataset is large enough. The training was on a single GPU. Typically it takes 3-5 days to show reasonable results, but it varies based on the target domain and the dataset size.