Skip to content

Commit

Permalink
Update docs/guides/haiku_migration_guide.rst
Browse files Browse the repository at this point in the history
Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com>
  • Loading branch information
cgarciae and 8bitmp3 committed Jul 14, 2023
1 parent ff9a9d9 commit 06583b5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/guides/haiku_migration_guide.rst
Original file line number Diff line number Diff line change
Expand Up @@ -791,7 +791,7 @@ we are telling ``nn.scan`` create different parameters for each step and slice t
return y

Notice how in Flax we pass ``None`` as the second argument to ``ScanBlock`` and ignore
its second output, these represent the inputs/outputs per-step but they are ``None``
its second output. These represent the inputs/outputs per-step but they are ``None``
because in this case we don't have any.

Initializing each model is the same as in previous examples. In this case,
Expand Down

0 comments on commit 06583b5

Please sign in to comment.