szymonmaszke / torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Mistake in Readme

va1shn9v opened this issue · comments

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