gluonts.torch.model.tft.lightning_module 模块

class gluonts.torch.model.tft.lightning_module.TemporalFusionTransformerLightningModule(model_kwargs: dict, lr: float = 0.001, patience: int =10, weight_decay: float = 0.0)

基类: lightning.pytorch.core.module.LightningModule

一个 pl.LightningModule 类，可用于使用 PyTorch Lightning 训练 TemporalFusionTransformerModel。

这是 (封装的) TemporalFusionTransformerModel 对象的一个薄层，它暴露了评估训练和验证损失的方法。

参数

• model_kwargs – 用于构建要训练的 TemporalFusionTransformerModel 的关键字参数。

• lr – 学习率。

• weight_decay – 权重衰减正则化参数。

• patience – 学习率调度器的耐心参数。

configure_optimizers()

返回要使用的优化器。

forward(*args, **kwargs)

与 torch.nn.Module.forward() 相同。

参数

• \*args – 您决定传递给 forward 方法的任何内容。

• \*\*kwargs – 关键字参数也是可能的。

返回

您模型的输出

training_step(batch, batch_idx: int)

执行训练步骤。

validation_step(batch, batch_idx: int)

执行验证步骤。