szymonmaszke / torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
https://szymonmaszke.github.io/torchlayers/
MIT License
568 stars 46 forks source link

Mistake in Readme #2

Closed va1shn9v closed 4 years ago

va1shn9v commented 4 years ago

Sorry to open an issue for such a trivial thing but in the readme, I think there is a small typo where after torchlayers.build() mnist model has conv1d layers and text classification model has conv2d layers