Skip to content

Apply optimizer every batch, not every epoch; unscale gradients before clipping#40

Merged
michaelmckinsey1 merged 2 commits into
LBANN:mainfrom
PatrickRMiles:miles30/minibatch_optimizer
Apr 2, 2026
Merged

Apply optimizer every batch, not every epoch; unscale gradients before clipping#40
michaelmckinsey1 merged 2 commits into
LBANN:mainfrom
PatrickRMiles:miles30/minibatch_optimizer

Conversation

@PatrickRMiles

Copy link
Copy Markdown
Collaborator

No description provided.

@PatrickRMiles PatrickRMiles changed the title Draft: apply optimizer every batch, not every epoch; unscale gradients before clipping Apply optimizer every batch, not every epoch; unscale gradients before clipping Apr 2, 2026
@michaelmckinsey1 michaelmckinsey1 self-requested a review April 2, 2026 20:26

@michaelmckinsey1 michaelmckinsey1 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This change decreases the necessary amount of epochs for convergence by roughly 5x

@michaelmckinsey1 michaelmckinsey1 merged commit 875f2fd into LBANN:main Apr 2, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants