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.

Sign up for GitHub

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

Jump to bottom

Merge dlwp-healpix updates from modulus-uw #785

Open
wants to merge 40 commits into
base: main
Choose a base branch
from
Open

Merge dlwp-healpix updates from modulus-uw #785

wants to merge 40 commits into from

Conversation

daviddpruitt
Copy link
Collaborator

Modulus Pull Request

Description

This PR adds additional features from the DLWP-Healpix model

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.
  • The CHANGELOG.md is up to date with these changes.
  • An issue is linked to this pull request.

Dependencies

nathanielcresswellclay and others added 30 commits November 22, 2024 18:13
@nathanielcresswellclay
Add workflow to automatically sync changes from nvidia/modulus into m…
3b15d51 
…ain branch
@zacespinosa
add WeightedOceanMSE to criterion
8f0cc02
@zacespinosa
Merge pull request #1 from AtmosSci-DLESM/WeightedOceanMSE
c439e00 
add WeightedOceanMSE to criterion
@zacespinosa
add optional gaussian noise to inputs and coupled variables during tr…
47fdce8 
…aining - should improve coupled stability
@zacespinosa
add random seed - still need to test
2826a02
@zacespinosa
remove datatransformer code - shouldn't be part of this PR
e44dcc9
@zacespinosa
move logging
f73e149
@nathanielcresswellclay
Removed blossom-ci workflow from modulus-uw fork, updated automatic sync
50041ef
@ivanauyeung
Fix the training and inference problem in nvidia modulus
e2f3376 
Signed-off-by: root <[email protected]>
@nathanielcresswellclay
Fix indexing in constant coupler
b8951be
@Bwformer
add 'Multi_SymmetricConvNeXtBlock'
20170ad
@Bwformer
Repalce 'n_layers' with 'n_conv_blocks' for clarity
68ca5bf
@Bwformer
Merge pull request #8 from AtmosSci-DLESM/fix_indexing_coupler
0c3288a 
Update indices in constant coupler.
@zacespinosa
Merge pull request #2 from AtmosSci-DLESM/GaussianNoiseCoupled
c81405e 
Gaussian Noise to Coupled Training
@ivanauyeung
Fix indexing in constant coupler
1317c02 
Signed-off-by: root <[email protected]>
@ivanauyeung
Fix indexing in constant coupler
027473b 
Signed-off-by: root <[email protected]>
@Bwformer
change back to 'n_layers' to match the old models
aff387e
@Bwformer
Merge pull request #9 from Bwformer/bw/DoubleConv
80821ec 
add 'Multi_SymmetricConvNeXtBlock'
@nathanielcresswellclay
enforce precedence of upstream modulus changes when auto syncing.
5726c7b
@nathanielcresswellclay
Merge pull request #3 from AtmosSci-DLESM/update_workflow
4574532 
Update workflow.
@yairchn
Merge branch 'main' into dlwp_coupled_training_inference_fix
be854fd
@yairchn
set scaling for mean: 0, std: 1 where no change is needed
957b00a
@yairchn
Merge branch 'dev' into dlwp_coupled_training_inference_fix
d519a84
@yairchn
Merge pull request #7 from AtmosSci-DLESM/dlwp_coupled_training_infer…
afb4e63 
…ence_fix

Fix the training and inference problem in nvidia modulus
@daviddpruitt
merge with upstream
84408ec
@daviddpruitt
merge with dev, coupler fixes
34be60b
@daviddpruitt
merge fixes and doc updates
9a19001
@daviddpruitt
Fix on batch size check, logging cleanup
5d8fab7
@daviddpruitt
test and lint fixes
cdadeae
@daviddpruitt
update coupler doc
5206808
@pzharrington pzharrington self-requested a review February 6, 2025 22:46
@pzharrington
Copy link
Collaborator

\blossom-ci

@pzharrington
Copy link
Collaborator

\blossom-ci

@NickGeneva
Copy link
Collaborator

\blossom-ci

@NickGeneva
Copy link
Collaborator

/blossom-ci

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@pzharrington pzharrington pzharrington approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
8 participants
@daviddpruitt @pzharrington @NickGeneva @nathanielcresswellclay @zacespinosa @ivanauyeung @Bwformer @yairchn