physicsnemo icon indicating copy to clipboard operation
physicsnemo copied to clipboard

🚀[FEA]: Add gradient clipping to StaticCaptureTraining performance optimization decorator

Open daviddpruitt opened this issue 1 year ago • 0 comments

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

daviddpruitt avatar May 14 '24 18:05 daviddpruitt