Nikolai10 / PerCo

PyTorch implementation of PerCo (Towards Image Compression with Perfect Realism at Ultra-Low Bitrates, ICLR 2024)
Apache License 2.0
53 stars 2 forks source link

AssertionError z_hat_indices_restored not equal z_hat_indices #3

Closed NiccoloNiccoli closed 4 months ago

NiccoloNiccoli commented 4 months ago

I tried running the Compression cell in PerceptualCompression.ipynb using the 0.0313bpp model. However, I get an AssertionError telling me that z_hat_indices_restored is not equal to z_hat_indices (at line 247 of compression_utils.py).

What am I doing wrong?

Nikolai10 commented 4 months ago

Dear @NiccoloNiccoli,

thanks for your interest!

We have been unable to reproduce the error. Have you updated the target-rate from target_rate = 0.1250 to target_rate = 0.0313?

NiccoloNiccoli commented 4 months ago

Thank you! I did not change the parameter before. Now that I have changed it, it works fine!