Skip to content

Fixes and updates for integrated neml2 training#15

Merged
reverendbedford merged 6 commits intomainfrom
weight_stochastic
Feb 14, 2025
Merged

Fixes and updates for integrated neml2 training#15
reverendbedford merged 6 commits intomainfrom
weight_stochastic

Conversation

@reverendbedford
Copy link
Collaborator

Adds a few miscellaneous features that help with training models over realistic data:

  1. Weighting in stochastic models
  2. Masking out bad tests from the loss calculation
  3. Important: fix nans in the gradient calculation.

@github-actions
Copy link

github-actions bot commented Feb 14, 2025

PR Preview Action v1.6.0
Preview removed because the pull request was closed.
2025-02-14 19:05 UTC

@reverendbedford reverendbedford merged commit fb34502 into main Feb 14, 2025
7 checks passed
@reverendbedford reverendbedford deleted the weight_stochastic branch February 14, 2025 19:02
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