TFT predict() with output_dir() running out of memory on large validation sets

sktime / pytorch-forecasting

Time series forecasting with PyTorch

https://pytorch-forecasting.readthedocs.io/

MIT License

3.98k stars 631 forks source link

Open pcgm-team opened 11 months ago

pcgm-team commented 11 months ago

It appears there is some sort of memory leak.

new_raw_predictions = model.predict(validation, mode="prediction", return_index = True, batch_size=128, output_dir = PATH)

shellrazer commented 5 months ago

same problem

IlIlllIIllIIlll commented 2 months ago

Me, too.