/visual-backprop

Pytorch implementation of VisualBackprop algorithm

Primary LanguagePythonMIT LicenseMIT

Visual Backprop PyTorch

Pytorch implementation of VisualBackprop algorithm: https://arxiv.org/pdf/1611.05418.pdf

This is MIPT Data Visualisation course project.

Main implementation (VisualBackpropWrapper) in visual_backprop.py

Example visualization in script visualize_mnist_training.py: visualizing regions of interests for LeNet network during MNIST classification training