gluonts.torch.model.deepar.lightning_module module

class gluonts.torch.model.deepar.lightning_module.DeepARLightningModule(model: gluonts.torch.model.deepar.module.DeepARModel, loss: gluonts.torch.modules.loss.DistributionLoss = NegativeLogLikelihood(), lr: float = 0.001, weight_decay: float = 1e-08)[source]

Bases: pytorch_lightning.core.lightning.LightningModule

configure_optimizers()[source]

Returns the optimizer to use

training = None
training_step(batch, batch_idx: int)[source]

Execute training step

validation_step(batch, batch_idx: int)[source]

Execute validation step