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

Per rank mlflow tracking location and updating fcn_afno config for easier benchmarking #121

Closed akshaysubr closed 1 year ago

akshaysubr commented 1 year ago

Modulus Pull Request

Description

Small PR that adds:

per rank mlflow tracking location to fix multi-gpu race conditions
small updates to fcn_afno config for easier benchmarking

Closes https://github.com/NVIDIA/modulus-launch/issues/66

Checklist

[x] I am familiar with the Contributing Guidelines.
[ ] The CHANGELOG.md is up to date with these changes.

Dependencies

NickGeneva commented 1 year ago

/blossom-ci

NickGeneva commented 1 year ago

/blossom-ci