allennlp.training.callbacks.update_momentum#

UpdateMomentum#

UpdateMomentum(self, momentum_scheduler:allennlp.training.momentum_schedulers.momentum_scheduler.MomentumScheduler) -> None

Callback that runs a Momentum Scheduler.

Parameters

  • momentum_scheduler : MomentumScheduler The momentum scheduler to run.