Skip to content

Commit

Permalink
Merge pull request #2018 from google:levskaya-docs-tweak-1
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 437971065
  • Loading branch information
Flax Authors committed Mar 29, 2022
2 parents 313014e + 8bed571 commit 85ac2ca
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions docs/design_notes/setup_or_nncompact.rst
Original file line number Diff line number Diff line change
Expand Up @@ -80,12 +80,6 @@ Reasons to prefer using ``setup``:
3. Allows defining more than one "forward pass" method
(see :class:`MultipleMethodsCompactError <flax.errors.MultipleMethodsCompactError>`)

**NOTE**: You can also define a module using both ``setup`` and
``nn.compact``. You'd put submodules and variables shared across
methods in ``setup``. Then, one method alone can be annotated
with ``nn.compact`` and add additional
submodules and variables that are not available in any other methods.




Expand Down

0 comments on commit 85ac2ca

Please sign in to comment.