search

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

Fix lr scheduler for case "if args.max_train_steps is None" #2

Closed Nikolai10 closed 4 months ago

Nikolai10 commented 4 months ago

See diffusers pull request fix(training): lr scheduler doesn't work properly in distributed scenarios

Current experiments are not affected by this issue (as we always set max_train_steps to 50k) -> should be however resolved in future releases.

Nikolai10 commented 4 months ago

fixed, see https://github.com/Nikolai10/PerCo/commit/c63142a65d7bf25e40e9ba68e748ac5de9e068ca