/esgd

ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.

Primary LanguagePythonMIT LicenseMIT

Stargazers