The reconstruction image degrade to the gray

[tianjiu233]

Hi, I have tried your code on a image dataset. However, after the training about 40 epochs, I tried the swin_mae on the image I used, image reconstructed by the model is always the gray one. Have you met the case and might it be solved? Hope you help and thanks for your time.