pytorchviz
pytorchviz copied to clipboard
Is this working properly with Conv2d and MaxPool2d?
I'm trying to use the library to visualize a CNN, but the output doesn't seem consistent. Thanks