search

XDZhelheim / STAEformer

[CIKM'23] Official code for our paper "Spatio-Temporal Adaptive Embedding Makes Vanilla Transformer SOTA for Traffic Forecasting".
https://arxiv.org/abs/2308.10425
132 stars 16 forks source link

非常好的代码 #4

Closed Suasy closed 6 months ago

Suasy commented 7 months ago

跑完了实验，感觉这个训练和模型的框架非常值得学习，写得非常的清晰明了👍

ownnaruto commented 7 months ago

同意，非常的简洁清晰，而且复现结果基本一致

XDZhelheim commented 7 months ago

感谢朋友们的认可哈哈，感兴趣的话可以关注我另一个公开项目，大号的训练框架。任何相关问题都欢迎交流！

Suasy commented 7 months ago

感谢朋友们的认可哈哈，感兴趣的话可以关注我另一个公开项目，大号的训练框架。任何相关问题都欢迎交流！

请问作者我可以在你的代码基础上进行修改和做我的工作吗，届时会添加引用和以及仓库说明，不知道可不可以🙏

XDZhelheim commented 7 months ago

当然可以。本来就是开源的，声明代码引用和论文引用就行。

Suasy commented 7 months ago

好的谢谢！