search

RobertTLange / evosax

Evolution Strategies in JAX 🦎
Apache License 2.0
475 stars 44 forks source link

Fix ParameterReshaper compatibility with dm-haiku #21

Closed vuoristo closed 2 years ago

vuoristo commented 2 years ago

Fix #20. ParameterReshaper uses / as a naming separator. This causes a clash with dm-haiku. This patch makes the separator configurable and changes the default separator to $.

RobertTLange commented 2 years ago

Thank you very much! That's a great PR -- most of my development is done with flax as the NN backend of choice, so this is greatly appreciated. Have a good weekend and let me know if you encounter any other issues, Rob