Wrong ms-ssim q4 pretrained model for cheng2020-anchor

[felixcheng97]

Thanks for your great work!

When I download the ms-ssim q4 pretrained model for cheng2020-anchor

https://github.com/InterDigitalInc/CompressAI/blob/278f45e6f444448d3c44d973e51f9723768afa18/compressai/zoo/image.py#L149

, it seems that the channel dimension of this checkpoint is 128, which should be 192 instead.

https://github.com/InterDigitalInc/CompressAI/blob/278f45e6f444448d3c44d973e51f9723768afa18/compressai/zoo/image.py#L215-L222

Could you help fix this pretrained model? Thank you very much.

[fracape]

Hi sorry for the late reply. Good eye, thanks. I'll try to find the time to correct this. From our plots in this repo and experience, that should not change the performance by much at this mid-range of bitrates.