TypeError: `model` must be a `LightningModule` or `torch._dynamo.OptimizedModule`, got `RecurrentNet
·
在训练我的时序预测模型的时候，想用pytorch_forecasting库中的RecurrentNetwork构建一个lstm模型，在使用pytorch_lightning库中的trtainer训练器训练我的模型时，在trainer.fit代码行报错显示TypeError: `model` must be a `LightningModule` or `torch._dynamo.OptimizedModule`, got `RecurrentNet，但是RecurrentNetwork它本身是继承了Lightnin Module的，于是我看到网上有人说pytorch_lightning库在更新后导入该模块的时候，之前的import lightning.pytorch要写成import pytorch_lightning，而RecurrentNetwork继承的AutoRegressiveBaseModelWithCovariates模型继承了Lightnin Module，也就是在basemodel.py文件中，我发现导入pytorch_lightning库的代码是import lightning.pytorch，改成import pytorch_lightning就好了
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
5
0- 0
已为社区贡献1条内容
所有评论(0)