sczhou/CodeFormer

Question on Training Steps and Cross entropy Loss for Stage 2

lxxie298 opened this issue · 2 comments

Your work is truly impressive! The clarity and depth you bring to the subject are remarkable!
Could you please let me know how many training steps are required in stage 2 and how small the Cross Entropy Loss should be to achieve normal images?

I use the pretrained VQGAN which provided by authors to start the training of stage-2, but it seem that it could generate a normal image. I have trained 70k step and batch size is 8, learning rate is 4e-5. following is my loss figure and results:
image
image

Please, kind-hearted person, help me. I don't know what the problem is.

Have you found the reason? The same problem arises

Have you found the reason? The same problem arises

No, I just run another experiments. If I have time later, I will look for the reason