search

alexlenail / NN-SVG

Publication-ready NN-architecture schematics.
http://alexlenail.me/NN-SVG/
MIT License
4.49k stars 579 forks source link

Feature request: Scaling for vectors in LeNet #39

Open thomasoliveira opened 3 years ago

thomasoliveira commented 3 years ago

I would just like to suggest the addition of a scaling for the Vector in the LeNet style schematics, or the ability to set the Vector Length and the text over the Vector separately.

For example, if a Dense layer connects 16@100x100 to 1x2048, the length of the vector looks too long for a printed figure.

alexlenail commented 3 years ago

I'd love to see this! Unfortunately I probably won't have time myself, but would look over a PR.