gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.learning_rate module

class gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.learning_rate.LearningRateScheduleCallback(milestones: List[float], decay: float = 0.5)

Bases: gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.base.Callback

The learning rate schedule callback decreases the learning rate by a predefined factor after each of the provided milestones (after x seconds during training).

on_train_batch_end(network: mxnet.gluon.block.HybridBlock, time_elapsed: float) → None

Hook called after every training batch.

on_train_start(trainer: mxnet.gluon.trainer.Trainer) → None

Hook called before training is run.