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

class gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.count.ParameterCountCallback[source]#

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

This callback allows counting model parameters during training.

num_parameters#

The number of parameters of the model. This attribute should only be accessed after training.

on_network_initialization_end(network: mxnet.gluon.block.HybridBlock) → None[source]#

Hook called once the network is initialized.