`model` 必须是 `LightningModule` 或 `torch._dynamo.OptimizedModule`，得到 `TemporalFusionTransformer`

Question

我试图从 pytorch_forecasting 模块进行时间融合变压器，但我在 trainer.fit 方法中遇到错误：model必须是 aLightningModule或torch._dynamo.OptimizedModule，得到TemporalFusionTransformer。我只是从“towardsdatascience”复制这篇论文。参考： https: //towardsdatascience.com/temporal -fusion-transformer-time-series-forecasting-with-deep-learning-complete-tutorial-d32c1e51cd91#:~:text=T%20emporal%20F%20usion%20T,dynamics%20of%20multiple%20time%20sequences。

Answer 1

需求已更新pytorch-forecasting，pytorch lighting 不再导入为lightning.pytorch, but pytorch_lightning。

改变这个pytorch-forecasting basemodel.py为我解决了这个问题。