How NBEATS stops writing numerous checkpoints files

unit8co / darts

A python library for user-friendly forecasting and anomaly detection on time series.

https://unit8co.github.io/darts/

Apache License 2.0

7.91k stars 857 forks source link

How NBEATS stops writing numerous checkpoints files #504

Closed YananLi18 closed 2 years ago

YananLi18 commented 2 years ago

Whenever I use NBEATS, hundreds of GB of checkpoint files are created. I also tried setting work_dir=None in the NBEATSModel(), but this will report an error.

microprediction commented 2 years ago

Oh my goodness I just discovered the same issue. Is there a workaround?

jonhilgart22 commented 2 years ago

have you tried the force_reset=True option?

hrzn commented 2 years ago

@lyn-bupt you closed the issue - did you find a way to address the issue in your case?

We still intend to work on improving and making checkpointing optional.