Add early stopping to model training

zalandoresearch / pytorch-ts

PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend

MIT License

1.21k stars 190 forks source link

Add early stopping to model training #45

Open larkz opened 3 years ago

larkz commented 3 years ago

Currently the loss function in model training only measures training loss per epoch. Add a feature to include validation loss as an early stopping criteria.