gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.base module#

class gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.base.Callback[source]#

Bases: object

A stripped-down callback which is focused on batches rather than epochs.

on_network_initialization_end(network: mxnet.gluon.block.HybridBlock) → None[source]#: Hook called once the network is initialized.

on_train_batch_end(network: mxnet.gluon.block.HybridBlock, time_elapsed: float) → None[source]#: Hook called after every training batch.

on_train_start(trainer: mxnet.gluon.trainer.Trainer) → None[source]#: Hook called before training is run.

on_validation_epoch_end(loss: float) → None[source]#: Hook called after every validation epoch.

class gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.base.CallbackList(callbacks: List[gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.base.Callback])[source]#

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

Wrapper class for a list of callbacks.

on_network_initialization_end(network: mxnet.gluon.block.HybridBlock) → None[source]#: Hook called once the network is initialized.

on_train_batch_end(network: mxnet.gluon.block.HybridBlock, time_elapsed: float) → None[source]#: Hook called after every training batch.

on_train_start(trainer: mxnet.gluon.trainer.Trainer) → None[source]#: Hook called before training is run.

on_validation_epoch_end(loss: float) → None[source]#: Hook called after every validation epoch.