search

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.17k stars 232 forks source link

Is it allowed to update the EntropyBottleneck while training? #253

Closed tuning12 closed 7 months ago

tuning12 commented 1 year ago

Thanks for your excellent work. I wonder if it is allowed to update the EntropyBottleneck while training. I still couldn't understand what "update" are doing.

YodaEmbedding commented 1 year ago

update fills in the discretized CDFs. Those are not used during training. They are used during compress and decompress. Just be sure to call update(force=True) immediately before each test/infer epoch, e.g., https://github.com/InterDigitalInc/CompressAI-Trainer/blob/master/compressai_trainer/runners/base.py#L90-L92

tuning12 commented 1 year ago

Thank you for your reply.