MS-SSIM Fine-tuning epochs
Indraa145 opened this issue · comments
Indraa145 commented
Hello, in the CompressAI documentation, it is mentioned that "MS-SSIM optimized networks were fine-tuned from pre-trained MSE networks (with a learning rate of 1e-5 for both optimizers)."
I want to ask how many training epochs did you do the fine-tuning for?
Also, while I finetune the model, it seems while the overall loss do get lower, my SSIM value only increase slightly, while my PSNR value keep going down as more iteration goes on. I use my own dataset though, not Vimeo90K.
Thank you.