Skip to content

Commit

Permalink
Merge pull request #15 from pitmonticone/main
Browse files Browse the repository at this point in the history
Update README.md
  • Loading branch information
rmsrosa authored Oct 22, 2023
2 parents 51f49c2 + 2b0b8a1 commit 59a4aef
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ There is, however, only partial support for multidimensional layers (convolution

## How it works

There is a distinction between netwoks starting with a layer with fixed-size input (Dense and Recurrent) and networks starting with a layer with variable-size input (Convolutional, Pooling, and functional).
There is a distinction between networks starting with a layer with fixed-size input (Dense and Recurrent) and networks starting with a layer with variable-size input (Convolutional, Pooling, and functional).

In the former case, just passing a network `m = Chain(...)` to plot works, e.g. `plot(m)`. In the latter case, one needs to pass along an initial input `inp`, or input size `inpsz = size(inp)`, as the second argument, like `plot(m, inp)` or `plot(m, inpsz)`, so that the plot recipe can properly figure out the size of each layer.

Expand Down

0 comments on commit 59a4aef

Please sign in to comment.