GAN_stability icon indicating copy to clipboard operation
GAN_stability copied to clipboard

Questions with batch normalization

Open LuChengTHU opened this issue 5 years ago • 1 comments

WechatIMG57 Hello, I just added BN in resblock and run 'python train.py configs/celebA-HQ', then it cannot converge at all.(see the picture below)

00006000

I noticed the moving average in your training, but I think its influence to BN is quite small. I cannot understand the reason why adding BN cannot converge. Looking forward to your reply!

LuChengTHU avatar Mar 27 '19 14:03 LuChengTHU

The moving average has a significant influence with batch normalization, as it simply ignores the learned weights from the BN when copying the model. Can you try to deactivate the moving average in the code and try again? Also note that our theoretical results are local in nature and we therefore cannot guarantee global convergence for every possible architecture you come up with :-) . However, from your image it appears that the algorithm is doing something sensible.

LMescheder avatar Mar 27 '19 15:03 LMescheder