CodeFormer
CodeFormer copied to clipboard
Question on Training Steps and Cross entropy Loss for Stage 2
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:
Please, kind-hearted person, help me. I don't know what the problem is.