Skip to content

feat(training): add validation split, early stopping, checkpointing, …#13

Merged
gaurav12301010 merged 1 commit intoOPCODE-Open-Spring-Fest:mainfrom
SUJALGOYALL:feature-lstm-validation
Oct 22, 2025
Merged

feat(training): add validation split, early stopping, checkpointing, …#13
gaurav12301010 merged 1 commit intoOPCODE-Open-Spring-Fest:mainfrom
SUJALGOYALL:feature-lstm-validation

Conversation

@SUJALGOYALL
Copy link
Copy Markdown
Contributor

Description

This pull request improves the LSTM model training loop by adding several enhancements to improve generalization, prevent overfitting, and stabilize learning.

Key Updates

  • Added a validation split (~15%) for monitoring validation metrics.
  • Implemented early stopping with best-weight restoration.
  • Added model checkpointing to automatically save the best model.
  • Applied ReduceLROnPlateau scheduler to dynamically adjust the learning rate when validation loss stalls.

These changes make the training process more efficient and robust while reducing overfitting.


Semver Changes

  • Patch (bug fix, no new features)
  • Minor (new features, no breaking changes)
  • Major (breaking changes)

Issues

Closes #3 >


Checklist

@gaurav12301010 gaurav12301010 merged commit f0561e8 into OPCODE-Open-Spring-Fest:main Oct 22, 2025
7 of 20 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Enhance Training Pipeline with Validation Split, EarlyStopping, Checkpointing and LR Scheduling

2 participants