Can ProbAttention in informer.py be directly replaced with AutoCorrelation of autoformer?

thuml / Autoformer

About Code release for "Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting" (NeurIPS 2021), https://arxiv.org/abs/2106.13008

MIT License

2k stars 429 forks source link

Can ProbAttention in informer.py be directly replaced with AutoCorrelation of autoformer? #82

Closed SQY2021 closed 2 years ago

SQY2021 commented 2 years ago

I want to replace probattention in informer.py with autocorrelation in autoformer, but I don't know if the model directly replaced is correct? I use the informer model in autoformer. https://github.com/thuml/Autoformer

微信截图_20220908111236

wuhaixu2016 commented 2 years ago

yes，you can replace this directly.