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

DLWP enhancements #94

Closed ktangsali closed 1 year ago

ktangsali commented 1 year ago

Modulus Pull Request

Description

This PR enables CUDA Graphs and AMP for this example. It also expands on the README for this example and also fixes a bug with the multi-gpu run.

Closes #93

Checklist

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

Dependencies

ktangsali commented 1 year ago

/blossom-ci

ktangsali commented 1 year ago

/blossom-ci

ktangsali commented 1 year ago

/blossom-ci

ktangsali commented 1 year ago

/blossom-ci