AutoTimes icon indicating copy to clipboard operation
AutoTimes copied to clipboard

Memory leakage in long term forecast

Open MyssH opened this issue 10 months ago • 1 comments

Description: I encountered a memory leakage issue when using a large dataset for long-term forecasting. It seems the problem occurred because the loss was being accumulated as a tensor instead of a scalar.

Solution: To fix this, I changed: loss_val += loss to loss_val += loss.item() at the following lines:

  • exp/exp_long_term_forecasting.py#144

  • exp/exp_long_term_forecasting.py#149

MyssH avatar Mar 27 '25 13:03 MyssH

Thanks for your feedback! We'll fix it :)

WenWeiTHU avatar Jul 22 '25 01:07 WenWeiTHU