Skip to content

Commit

Permalink
fixup
Browse files Browse the repository at this point in the history
  • Loading branch information
mcabbott committed Oct 10, 2022
1 parent 66d7dc4 commit 23baf23
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/src/models/basics.md
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ Flux.@functor Affine
Finally, most Flux layers make bias optional, and allow you to supply the function used for generating random weights. We can easily add these refinements to the `Affine` layer as follows:

```
function Affine((in, out)::Pair; bias = true, init=Flux.randn32)
function Affine((in, out)::Pair; bias=true, init=Flux.randn32)
W = init(out, in)
b = Flux.create_bias(W, bias, out)
Affine(W, b)
Expand All @@ -231,7 +231,7 @@ end
Affine(3 => 1, bias=false, init=ones) |> gpu
```

```@doc
```@docs
Functors.@functor
Flux.create_bias
```

0 comments on commit 23baf23

Please sign in to comment.