Could you support a pre-trained model?

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

Closed yanshuaibupt closed 1 year ago

yanshuaibupt commented 1 year ago

Thanks for your excellent job, I cost a lot of time on training the model on my device. Could you upload a pre-trained model?

aptx1231 commented 1 year ago

抱歉回复的比较晚，才看到issue。我提供pems04和nyxtaxi的预训练模型。注意论文的结果是多次实验的平均值，下边提供的只是一次实验产出的预训练模型哦。