Modernization - Githubissues

facebookresearch / schedule_free

Schedule-Free Optimization in PyTorch

Apache License 2.0

1.91k stars 65 forks source link

Closed adefazio closed 3 weeks ago

adefazio commented 3 weeks ago

Change the code to follow more modern PyTorch design: i.e. removing .data calls and adding in no_grad annotations
Add more type annotations
Addresses stability issues from #38

I'm going to run some tests of this to ensure it gives similar results to the published version of the method before merging.