Justin-Tan / high-fidelity-generative-compression

Pytorch implementation of High-Fidelity Generative Image Compression + Routines for neural image compression
Apache License 2.0
411 stars 77 forks source link

Fix typo in learning rate decay logging. #53

Open Marvel0usx opened 1 year ago

Marvel0usx commented 1 year ago

Hello Justin, I hope you're doing well. I came across a small typo that caused the entire training process to halt. It could potentially waste several iterations of training.

The error is triggered when a new Adam optimizer setup (other than the one in the pre-trained model) is provided and occurs when the learning rate is updated.

It is truly amazing to find a PyTorch implementation of HiFiC, and I appreciate your time and efforts in maintaining this project!