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 jax.numpy as jnp
|
||||
|
||||
class MLPMoudle(nn.Module):
|
||||
class MLPModule(nn.Module):
|
||||
config: MLPConfig
|
||||
dtype: jnp.dtype = jnp.float32
|
||||
|
||||
|
||||
Reference in New Issue
Block a user