NVIDIA / modulus-launch

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems

Apache License 2.0

54 stars 27 forks source link

Add Stokes flow example #107

Closed sifanexisted closed 10 months ago

sifanexisted commented 1 year ago

Modulus Pull Request

Description

This PR includes the training and inference scripts for the example of learning Stokes flow with MeshgraphNet. Also fix a small bug in the checkpointing.

Checklist

[x] I am familiar with the Contributing Guidelines.
[] New or existing tests cover these changes.
[x] The documentation is up to date with these changes.
[ ] The CHANGELOG.md is up to date with these changes.

Dependencies

mnabian commented 1 year ago

/blossom-ci

mnabian commented 1 year ago

/blossom-ci

mnabian commented 11 months ago

/blossom-ci

NickGeneva commented 10 months ago

Closing since PR is here: https://github.com/NVIDIA/modulus/pull/219