Skip to content

Fix AdaBelief variance updates#55

Open
linrock wants to merge 1 commit intolessw2020:mainfrom
linrock:fix-adabelief-variance
Open

Fix AdaBelief variance updates#55
linrock wants to merge 1 commit intolessw2020:mainfrom
linrock:fix-adabelief-variance

Conversation

@linrock
Copy link
Copy Markdown

@linrock linrock commented Apr 19, 2026

Fixes these calculations when AdaBelief is enabled:

  • use addcmul_ so variance_ma_belief is updated in place
  • use variance_ma_belief instead of variance_ma in AdaBelief variance calculations

Fixes these calculations when AdaBelief is enabled:
- use addcmul_ so variance_ma_belief is updated in place
- use variance_ma_belief instead of variance_ma in AdaBelief
  variance calculations
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.

1 participant