InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research
https://interdigitalinc.github.io/CompressAI/
BSD 3-Clause Clear License
1.15k stars 228 forks source link

training SSF2020 model with quality 8 #282

Open Maryamsana-1998 opened 4 months ago

Maryamsana-1998 commented 4 months ago

Bug

I have been trying to train ssf2020 model with vimeo90k dataset with quality 8 to compare it with the tests reported in tests folder. but i am not getting comparable results for it. Results I'm getting

image

To Reproduce

  1. change the file examples/train_video.py
    net = video_models[args.model](quality=8) and train_dataset.max_frames = 8 test_dataset.max_frames = 8
  2. now train this model with command
    python3 examples/train_video.py -d '/data/maryam.sana/vimeo_septuplet/' --cuda

Can you please tell me what should be done to train a quality 8 model correctly? what changes do I have to make?

fracape commented 4 months ago

is your problem specific to that operation point (quality=8)?