physicsnemo
physicsnemo copied to clipboard
🚀[FEA]: Add gradient clipping to StaticCaptureTraining performance optimization decorator
Is this a new feature, an improvement, or a change to existing functionality?
Improvement
How would you describe the priority of this feature request
Low (would be nice)
Please provide a clear description of problem you would like to solve.
Add gradient clipping support to the performance optimization decorator. The performance optimization decorator doesn't support Gradient Clipping. This prevents its use for applications that require gradient clipping.
Describe any alternatives you have considered
No response