search

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

Is this working properly with Conv2d and MaxPool2d? #30

Open rami-alloush opened 5 years ago

rami-alloush commented 5 years ago

I'm trying to use the library to visualize a CNN, but the output doesn't seem consistent. Thanks