Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
More correct hessian trace calculation (#2159)
### Changes Use absolute value of avg_total_trace in denominator, as done in [PyHessian:](https://github.com/amirgholami/PyHessian/blob/master/pyhessian/hessian.py#L186C22-L186C22) ### Reason for changes more robust mixed precision algo ### Related tickets resolves #2155 ### Tests hawq-related tests
- Loading branch information