Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Development #9

Merged
merged 5 commits into from
Sep 21, 2023
Merged

Development #9

merged 5 commits into from
Sep 21, 2023

Conversation

dfulu
Copy link
Member

@dfulu dfulu commented Sep 21, 2023

Pull Request

Description

  • Add stochastic weight averaging callback which will work with iterdatapipe dataset
  • Add factor to option to scale by capacity so the input values stay close to 1. This is important for weight decay to work properly across the network and for stable training.

Checklist:

  • My code follows OCF's coding style guidelines
  • I have performed a self-review of my own code
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works
  • I have checked my code and corrected any misspellings

@dfulu dfulu merged commit 3791075 into main Sep 21, 2023
2 checks passed
@peterdudfield peterdudfield deleted the development branch December 19, 2023 10:29
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