LevenbergMarquardt do not seems to work with non-flat input.
Open
- Dominant language
- Python
- Stars
- 1.1k
- Forks
- 76
- Avg merge
- 2d 21h
- Merged PRs (30d)
- 1
Description
GaussNewton is working as intended with Pytree. I would expect the same for LM. Instead, I had to flatten the array to make it properly works.
The errors appear at line 445. The error comes from the fact that the pytree of params and vec do not match.
Contributor guide
Assessment
This issue has not been assessed yet.