Visualisation tool for CNNs in pytorch - View it on GitHub