Finetune on my own dataset

[Solidays-Hung]

Thanks for your work! I load your pretrained R&D ckpt and finetune the model on my dataset. But after 40 epochs, the model seems to collapse. The Recon Loss and Restoration Loss suddenly increase. Besides, the result_dict['recon'] seems to be All black images. Could you tell me the possible cause of this?