fixed typo in flax-projects readme (#12466)
This commit is contained in:
@@ -502,7 +502,7 @@ First, write a Flax module that will declare the layers and computation.
|
|||||||
import flax.linen as nn
|
import flax.linen as nn
|
||||||
import jax.numpy as jnp
|
import jax.numpy as jnp
|
||||||
|
|
||||||
class MLPMoudle(nn.Module):
|
class MLPModule(nn.Module):
|
||||||
config: MLPConfig
|
config: MLPConfig
|
||||||
dtype: jnp.dtype = jnp.float32
|
dtype: jnp.dtype = jnp.float32
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user