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
901 stars 168 forks source link

当运行到训练的循环时，内存消耗骤增 #57

Closed sino19840626 closed 4 months ago

sino19840626 commented 4 months ago

当训练开始到for i, (batch_x, batch_y, batch_x_mark, batch_y_mark) in enumerate(train_loader):，内存消耗量骤增。为什么会这样？ 我做其他训练更大的数据训练都没出现过这种情况。