gluonts.torch.model.wavenet.lightning_module module#
- class gluonts.torch.model.wavenet.lightning_module.WaveNetLightningModule(model_kwargs: dict, lr: float = 0.001, weight_decay: float = 1e-08)[source]#
Bases:
lightning.pytorch.core.module.LightningModule
LightningModule wrapper over WaveNet.
- Parameters
model_kwargs – Keyword arguments to pass to WaveNet.
lr – Learning rate, by default 1e-3
optional – Learning rate, by default 1e-3
weight_decay – Weight decay, by default 1e-8
optional – Weight decay, by default 1e-8