Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.
And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.
Examples for traverse_util module
See original GitHub issueFirst off, loving the new linen api 🎉!!!
The traverse_util module looks useful for manipulating parameters/state, but I could not figure out how to use it effectively (apart from unflatten_dict and flatten_dict, which are both extremely useful). The simple examples in the docs make sense, but it’d be nice to see how to use this in a real setting. Perhaps the power of this module can be illustrated in a transfer learning example, where you need to manipulate subsets of parameters?
The module seems similar to Haiku’s data_structures module, but the API seems more complex.
Also, it’s unfortunate that Flax has jax_utils (plural) and traverse_util (singular). Is it too late to pluralize traverse_util for consistency?
Issue Analytics
- State:
- Created 3 years ago
- Comments:10 (5 by maintainers)
Top Results From Across the Web
Top Related Medium Post
Top Related StackOverflow Question
Troubleshoot Live Code
Top Related Reddit Thread
Top Related Hackernoon Post
Top Related Tweet
Top Related Dev.to Post
Top Related Hashnode Post
Yes, but maybe we should create a new one for removing the traverse_util API. Since you started this issue as well, do you want to do that?
Got it, I agree that it makes sense to remove this API altogether. I think the equivalent tree_map code is easier to understand as well. In this case, should I close this issue?