NVIDIA / modulus-launch

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems
Apache License 2.0
56 stars 27 forks source link

Update SFNO Recipe to align with updated SFNO model in modulus #125

Closed daviddpruitt closed 1 year ago

daviddpruitt commented 1 year ago

Modulus Pull Request

Description

Updates the SFNO training and inference recipes to align with updated SFNO model

Checklist

Dependencies

No new dependiencies.

mnabian commented 1 year ago

/blossom-ci

mnabian commented 1 year ago

@daviddpruitt Should we update the configs as well to include the best configs?

daviddpruitt commented 1 year ago

@mnabian That's a good idea. We'll need to check with NVR to see what the current best is.

mnabian commented 1 year ago

Closing as SFNO recipe will be updated in another PR: https://github.com/NVIDIA/modulus/pull/232