CodeFormer icon indicating copy to clipboard operation
CodeFormer copied to clipboard

Question on Training Steps and Cross entropy Loss for Stage 2

Open lxxie298 opened this issue 9 months ago • 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.

lxxie298 avatar May 23 '24 09:05 lxxie298