stargan-v2 icon indicating copy to clipboard operation
stargan-v2 copied to clipboard

Training results cannot meet expectations

Open jishulinbyte opened this issue 3 years ago • 6 comments

Hello, I used your settings to train on AFHQ and set the batchsize to 4. When switching to wild animals, the result was very poor. The FID index was still 60 at 80,000 steps.Can you help me answer my doubts? Thank you, I am very puzzled, because I want to conduct research based on your network, but the reproduction effect is not good, and I cannot proceed downward.

jishulinbyte avatar Jun 05 '21 03:06 jishulinbyte

发现是因为batchsize的问题

jishulinbyte avatar Jun 06 '21 03:06 jishulinbyte

Hi could you please share the hypeparameter you have used to avoid this problem (i.e. the batch size)? Thank you

LuigiSigillo avatar Sep 15 '21 10:09 LuigiSigillo

Batch size must be 8,Only then can we get the result with very low FID value,The batchsize is 4, the effect will be very poor

jishulinbyte avatar Sep 16 '21 08:09 jishulinbyte

So you only changed the batch size with respect to the default parameters ?

LuigiSigillo avatar Sep 16 '21 08:09 LuigiSigillo

yes

jishulinbyte avatar Sep 16 '21 14:09 jishulinbyte

When batch=4, training will become unstable.

jishulinbyte avatar Sep 16 '21 14:09 jishulinbyte