From 75f67e8e516b080278f805d70132bb584d7218a8 Mon Sep 17 00:00:00 2001 From: Niklas Schmitz Date: Fri, 27 May 2022 13:41:00 +0200 Subject: [PATCH] Fix optax import in readme --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index a235dc34..1e91e3c2 100644 --- a/README.md +++ b/README.md @@ -68,7 +68,7 @@ class MLP(eg.Module): **2.** Create a `Model` from this module and specify additional things like losses, metrics, and optimizers: ```python -import optax optax +import optax import elegy as eg model = eg.Model( @@ -106,7 +106,7 @@ To use Flax with Elegy just create a `flax.linen.Module` and pass it to `Model`. ```python import jax import elegy as eg -import optax optax +import optax import flax.linen as nn class MLP(nn.Module): @@ -149,7 +149,7 @@ You can also optionally create your own `hk.Module` and use it in `forward` if n ```python import jax import elegy as eg -import optax optax +import optax import haiku as hk