This line
p.data.addcdiv_(-group['lr'], m, h.add(group['eps']).clamp(max=group['rho']))
should be changed to the following to match the paper
update = m.div(h.add(group['eps'])).clamp(max=group['rho'])
p.data.add_(update, alpha=-group['lr'])
kyegomez / Sophia
Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.
378
Contributors get 40% of received funds after fees
How does funding with Polar work?
1
Pay now to fund the work behind this issue.
2
Get updates on progress being made.
3
Maintainer is rewarded once the issue is completed.
FAQ
Backer
You're funding impactful open source efforts
Contributor
You want to contribute to this effort
Maintainer
You want to get funding like this too