Add `freeze!`/`thaw!` - Githubissues

FluxML / Optimisers.jl

Optimisers.jl defines many standard optimisers and utilities for learning loops.

https://fluxml.ai/Optimisers.jl

MIT License

72 stars 20 forks source link

Add `freeze!`/`thaw!` #112

Closed mcabbott closed 1 year ago

mcabbott commented 1 year ago

Replaces #49 with something much simpler: now that Leaf is mutable, you can simply pass it the appropriate limb of the state tree, rather than having some explicit address mechanism.

Addresses some of #107

PR Checklist

[x] Tests are added
[x] Documentation, if applicable

mcabbott commented 1 year ago

Shall we do this?

Would be nice to include in https://github.com/FluxML/Flux.jl/pull/2114 while sorting that out.