Model visualization #4
Description
Issues while building massive projects are mostly due to misinterpretation of what the model is and how gradients are flowing thru it. A Visualization might help here.
Example: # Add visualization for model like https://github.com/szagoruyko/functional-zoo/blob/master/resnet-18-export.ipynb