search

chengtan9907 / OpenSTL

OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning
https://openstl.readthedocs.io/en/latest/
Apache License 2.0
776 stars 129 forks source link

early stopping #146

Closed IsmailBData closed 3 months ago

IsmailBData commented 3 months ago

Hi,

how can i specify early stopping?

Thanks

chengtan9907 commented 3 months ago

PyTorch lightning itself supports early stopping. You may add a callback in api/exp.py. The early stopping callback can be referred in lightning docs.

chengtan9907 commented 3 months ago

I will close this issue if there are no more questions. Please feel free to open a new issue when you have new questions.