Giters

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

Is it allowed to update the EntropyBottleneck while training?

tuning12 opened this issue · comments

commented

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.

commented

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

commented

Thank you for your reply.