search

LS4GAN / uvcgan2

UVCGAN v2: An Improved Cycle-Consistent GAN for Unpaired Image-to-Image Translation
https://arxiv.org/abs/2303.16280
Other
116 stars 18 forks source link

Batch size #31

Open SASasfc opened 1 month ago

SASasfc commented 1 month ago

Hello, I would like to ask about the training process, the batchsize is set to 1 to train 500epoch, I think the training time is a bit long, can I increase the batchsize to improve the training speed, but does this have an impact on the training results? Thank you!

usert5432 commented 1 month ago

Hi @SASasfc,

can I increase the batchsize to improve the training speed, but does this have an impact on the training results?

Based on my experience with CycleGAN-like models, increasing the batch size does not improve the training speed or the quality of the translations in most cases. In other words, generally one needs to train for 1M batches regardless of the batch size to get the same translation quality. However, the impact of the batch size can vary depending on your specific dataset. So, it might be beneficial to try increasing the batch size for your dataset to see if it helps.