InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

Home Page:https://interdigitalinc.github.io/CompressAI/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MS-SSIM Fine-tuning epochs

Indraa145 opened this issue · comments

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.