Bjarten/early-stopping-pytorch

Health v2 · maintenance only

Adoption not part of the score

1266 stars · 288 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A small PyTorch utility package providing an EarlyStopping class that monitors validation loss during training and stops when it stops improving. It also saves model checkpoints whenever validation loss decreases.

Use cases

• stop pytorch training when validation loss stops improving
• prevent overfitting in pytorch model training
• save best model checkpoint during training
• add early stopping with patience to training loop
• regularize deep learning model training

When to choose

• you train PyTorch models and want simple early stopping with checkpointing
• you want a lightweight pip-installable utility rather than a full training framework

When to avoid

• you use TensorFlow/Keras, which has built-in early stopping callbacks
• you need a full experiment tracking or training framework like PyTorch Lightning or Ignite

Facets

library · maturity stable

machine-learning developer-tools machine-learning deep-learning python pytorch early-stopping regularization model-training checkpointing

Member repositories

For agents

markdown · JSON · MCP: product_card(name="Bjarten/early-stopping-pytorch")