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

Logger Updates #120

Closed NickGeneva closed 1 year ago

NickGeneva commented 1 year ago

Modulus Pull Request

Description

Fixing up some issues with the loggers including:

Adding average iteration timing to all outputs
Only rank 0 logs
Fixing file log bug

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

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.

Dependencies