如何修复 Pytorch-Forecasting 模型拟合关于序列元素的 ValueError

Posted 2023-03-27

【中文标题】如何修复 Pytorch-Forecasting 模型拟合关于序列元素的 ValueError

【英文标题】：How do I fix Pytorch-Forecasting model fit ValueError about sequence element

【发布时间】：2021-11-02 11:12:18

【问题描述】：

我是 Pytorch_Forecasting 的新手。我完全按照“使用 Temporal Fusion Transformer 进行需求预测”(https://pytorch-forecasting.readthedocs.io/en/latest/tutorials/stallion.html) 中的描述进行操作。直到最后一个模型拟合步骤（trainer.fit(...)），一切都找到了。我不断收到错误消息，“ValueError：字典更新序列元素 #0 的长度为 1；需要 2”。我不知道这是在哪里发生的......我将不胜感激任何帮助、任何线索、任何解决我的问题的方法

谢谢！！

【参考方案1】：

我遇到了同样的问题。 pytorch_lightning，最近推出了一个新版本，pytorch-forecasting就是建立在它之上的。我将 torchmetrics 的版本更改为 0.5.0。 pip install torchmetrics==0.5.0

【讨论】：

不幸的是，这并没有为我解决。我有以下库。 pytorch-forecasting==0.9.0, pytorch-lightning==1.4.8, torch==1.9.1, torchmetrics==0.5.1

我认为您需要将 torchmetrics 的版本从 0.5.1 更改为 0.5.0

我尝试降级，但没有帮助。

【参考方案2】：

感谢this github 线程，我想我已经找到了解决方案。您需要的是以下库的两个特定版本：

pip install pandas==1.2.5
pip install torchmetrics==0.5.0

我尝试在 python 3.6 中安装 pandas==1.2.5 但无法做到。它被困在pandas==1.1.5。所以，我不得不使用 python 3.8。完整的相关包是：

pytorch-forecasting==0.9.0
pytorch-lightning==1.4.8
torch==1.9.1
torchmetrics==0.5.0
pandas==1.2.5