Lightning AI Studios: Never set up a local environment again →

← Back to blog

Avoid Overfitting in Machine Learning

Lightning EarlyStopping Callback

Using the EarlyStopping callback, Lightning will automatically stop training once a specific metric has stopped improving. This helps you avoid overfitting.

The default behavior is that EarlyStopping will happen once there hasn’t been an improvement in 3 evaluations. You can change this using the patience value.

Watch the video to learn more.