search

sczhou / CodeFormer

[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer
Other
15.95k stars 3.36k forks source link

Question on Training Steps and Cross entropy Loss for Stage 2 #373

Open lxxie298 opened 6 months ago

lxxie298 commented 6 months ago

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.

qinghaoya commented 6 months ago

Have you found the reason? The same problem arises

lxxie298 commented 6 months ago

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