search

thuml / iTransformer

Official implementation for "iTransformer: Inverted Transformers Are Effective for Time Series Forecasting" (ICLR 2024 Spotlight), https://openreview.net/forum?id=JePfAI8fah
https://arxiv.org/abs/2310.06625
MIT License
1.17k stars 206 forks source link

Memory Footprint? #56

Closed Atik-Ahamed closed 6 months ago

Atik-Ahamed commented 7 months ago

Dear Authors, Your research is highly interesting. It was a great finding for time-series forecasting. We are very interested in this. However, regarding the memory footprint mentioned in figure 10 of the paper, we are a bit confused about how it is measured. Is there any code available to measure this? Or can you provide some instructions on how we can measure this? For example, does it include GPU-only memory consumption or GPU+RAM memory consumption?

Thank you in advance.

WenWeiTHU commented 6 months ago

Hello, we measure the GPU-only memory consumption under the stated configuration in our paper (See Appendix for the details).