gluonts.torch.model.tide.lightning_module module#
- class gluonts.torch.model.tide.lightning_module.TiDELightningModule(model_kwargs: dict, lr: float = 0.001, weight_decay: float = 1e-08, patience: int = 10)[source]#
Bases:
lightning.pytorch.core.module.LightningModule
A
pl.LightningModule
class that can be used to train aTiDEModel
with PyTorch Lightning.This is a thin layer around a (wrapped)
TiDEModel
object, that exposes the methods to evaluate training and validation loss.- Parameters
model_kwargs – Keyword arguments to construct the
TiDEModel
to be trained.lr – Learning rate.
weight_decay – Weight decay regularization parameter.
patience – Patience parameter for learning rate scheduler.