search

poets-ai / elegy

A High Level API for Deep Learning in JAX
https://poets-ai.github.io/elegy/
MIT License
469 stars 32 forks source link

Develop root mean squared error metric #208

Closed anvelezec closed 2 years ago

anvelezec commented 2 years ago

Develop root mean squared error metric in treex: https://github.com/cgarciae/treex/issues/38