NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

https://developer.nvidia.com/modulus

Apache License 2.0

1.03k stars 243 forks source link

Add gradient clipping to Static Capture Utilities #660

Closed dallasfoster closed 2 months ago

dallasfoster commented 2 months ago

Modulus Pull Request

Description

Adds gradient_clip_norm to the static capture utilities. This should be backward compatible with existing examples.

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.
[ ] An issue is linked to this pull request.

dallasfoster commented 2 months ago

/blossom-ci

dallasfoster commented 2 months ago

/blossom-ci

dallasfoster commented 2 months ago

/blossom-ci

dallasfoster commented 2 months ago

/blossom-ci

dallasfoster commented 2 months ago

/blossom-ci

dallasfoster commented 2 months ago

/blossom-ci