search

daniel-j-h / nedem

Neural implicit digital elevation model
MIT License
2 stars 0 forks source link

Upstream l1 loss to optax #9

Open daniel-j-h opened 2 years ago

daniel-j-h commented 2 years ago

When trying out various losses I ran into optax providing l2 but not l1 loss. Should be a quick upstream pull request.

https://github.com/deepmind/optax/blob/252d152660300fc7fe22d214c5adbe75ffab0c4a/optax/_src/loss.py