gluonts.nursery.tsbench.src.tsbench.gluonts package

class gluonts.nursery.tsbench.src.tsbench.gluonts.TimedTrainer(training_time: float, validation_milestones: Optional[List[float]] = None, learning_rate: float = 0.001, callbacks: Optional[List[gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.base.Callback]] = None)

Bases: gluonts.mx.trainer._base.Trainer

A custom trainer whose training duration is based on wall clock time instead of epochs.

Subpackages

• gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks package
  • Submodules
    • gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.base module
    • gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.count module
    • gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.learning_rate module
    • gluonts.nursery.tsbench.src.tsbench.gluonts.callbacks.save module

Submodules

• gluonts.nursery.tsbench.src.tsbench.gluonts.trainer module