diff --git a/docs/guides/flax_fundamentals/arguments.md b/docs/guides/flax_fundamentals/arguments.md index 02f65e2eb..6e4ce98a3 100644 --- a/docs/guides/flax_fundamentals/arguments.md +++ b/docs/guides/flax_fundamentals/arguments.md @@ -4,7 +4,7 @@ In Flax Linen we can define `Module` arguments either as dataclass attributes or as arguments to methods (usually `__call__`). Typically the distinction is clear: -* Completely fixed properties, such as the choice of kernel initializer or number of output features, are hyperparameters and should be defined as dataclass attributes. Typically two Module instances with different hyperparamaters cannot share in a meaningful way. +* Completely fixed properties, such as the choice of kernel initializer or number of output features, are hyperparameters and should be defined as dataclass attributes. Typically two Module instances with different hyperparameters cannot share in a meaningful way. * Dynamic properties, such as input data and top-level "mode switches" like `train=True/False`, should be passed as arguments to `__call__` or another method. Some cases are however less clear cut. Take for example the `Dropout` module. diff --git a/flax/core/lift.py b/flax/core/lift.py index 7500c65bd..678110029 100644 --- a/flax/core/lift.py +++ b/flax/core/lift.py @@ -1171,7 +1171,7 @@ def cond( Note that this constraint is violated when creating variables or submodules in only one branch. Because initializing variables in just one branch - causes the paramater structure to be different. + causes the parameter structure to be different. Example::