BUAABIGSCity / PDFormer

[AAAI2023] A PyTorch implementation of PDFormer: Propagation Delay-aware Dynamic Long-range Transformer for Traffic Flow Prediction.
MIT License
209 stars 35 forks source link

关于使用PeMS07 出现killed问题 #20

Closed klayc-gzl closed 1 year ago

klayc-gzl commented 1 year ago

您好,我使用PeMS07进行训练,但是会出现killed,这应该是超内存了吧,我的内存只有24G,请问有没有什么办法能够调整一下PeMS 07的输入呢

klayc-gzl commented 1 year ago

您在pdformer中进行对比的其它模型可以通过libcity复现吗

aptx1231 commented 1 year ago

您好,我使用PeMS07进行训练,但是会出现killed,这应该是超内存了吧,我的内存只有24G,请问有没有什么办法能够调整一下PeMS

24G可能有点费劲,可以试试设置add-day-in-week是false,损失部分精度。

aptx1231 commented 1 year ago

您在pdformer中进行对比的其它模型可以通过libcity复现吗

可以的,运行libcity就行了。

klayc-gzl commented 1 year ago

好的好的,谢谢您,还有一个问题是请问您论文所给出的结果都是用的MASK的值吗?

aptx1231 commented 1 year ago

对的,这个是通用做法。

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: klayc-gzl @.> 发送时间: 2023年10月18日 19:04 收件人: BUAABIGSCity/PDFormer @.> 抄送: Jiawei Jiang @.>, Comment @.> 主题: Re: [BUAABIGSCity/PDFormer] 关于使用PeMS07 出现killed问题 (Issue #20)

好的好的,谢谢您,还有一个问题是请问您论文所给出的结果都是用的MASK的值吗?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

klayc-gzl commented 1 year ago

不好意思,还想再请教一下,因为您在add_time_in_day中其实是增加一个维度的时间信息,但是周期信息用的one-hot,然后现在我不是内存不太够吗,我就想将week也放进一个维度用0-6表示 image image 然后我看它维度确实变成3了: image 但是最后结果会出现一种奇怪的错误: image 太感谢您了!!!

klayc-gzl commented 1 year ago

将维度修改后不知道为什么会出现这种错误 image

klayc-gzl commented 1 year ago

和下面这种错误 image

klayc-gzl commented 1 year ago

已解决,非常感谢作者!