fixed typo in flax-projects readme (#12466)

This commit is contained in:
Matthew LeMay
2021-07-02 02:57:39 -04:00
committed by GitHub
parent e52288a140
commit b4ecc6bef2

View File

@@ -502,7 +502,7 @@ First, write a Flax module that will declare the layers and computation.
import flax.linen as nn
import jax.numpy as jnp
class MLPMoudle(nn.Module):
class MLPModule(nn.Module):
config: MLPConfig
dtype: jnp.dtype = jnp.float32