Fix bug in rewinding LR scheduling

IntelLabs / Model-Compression-Research-Package

A library for researching neural networks compression and acceleration methods.

Apache License 2.0

136 stars 24 forks source link

Closed ofirzaf closed 2 years ago

ofirzaf commented 2 years ago

Rewinding LR scheduling didn't give the correct factor in the interval between the end of warmup steps and the beginning of pruning.

This PR fixes this bug