Skip to content

v4.1.2

Choose a tag to compare

@shyuep shyuep released this 26 Oct 10:02
· 26 commits to main since this release
6eb8675
  • Fix: LR scheduler stepped twice per epoch. MatglLightningModuleMixin.on_train_epoch_end manually called
    scheduler.step() on top of Lightning's automatic epoch-interval step, so schedules ran at twice the intended
    rate (e.g. CosineAnnealingLR reached eta_min after decay_steps / 2 epochs and then rose again). The manual
    step is removed and decay_steps now means what it says. Behavior change: to reproduce runs trained with
    earlier versions, halve decay_steps (or the equivalent parameter of a custom scheduler). (#848)