maltesilber

Stepwise LR-Scheduler not working across epochs

3

### Bug description ## Description I'm training a model based on number of iterations instead of a number of epochs. The same model trains on datasets of different sizes, hence...

In the paper it states that the latent space get scaled based on the variance of the first batch. See: However this behaviour does not seem to be implemented, as...