search

moraieu / query-selector

LONG-TERM SERIES FORECASTING WITH QUERYSELECTOR – EFFICIENT MODEL OF SPARSEATTENTION
Apache License 2.0
74 stars 19 forks source link

memory explosion（Not gpu memory） #3

Open TruthK opened 2 years ago

TruthK commented 2 years ago

Hello, your work is excellent. However, when I used the electricity dataset (https://github.com/laiguokun/multivariate-time-series-data), a memory explosion problem occurred. I don't know if there is a solution. When running the EETh data set, it is normal. Thank You.

jackli7 commented 2 years ago

Hi! We actually never really applied our algorithm to this dataset but it is quite obvious that this dataset is much larger than ETT so it requires more GPU. Additionally, this dataset has a different structure so it requires some work to adjust to this new setup. However, we believe that if you implement those changes it should work in this case as well. We would appreciate if you let us know if you succide with this quest. Cheers and good luck.