/sps-pytorch

:eight_spoked_asterisk: An unofficial implementation of Stochastic Polyak Step-size in PyTorch

Primary LanguagePythonMIT LicenseMIT

Stargazers