PyTorch implementation of LeNet-5 with live visualization - View it on GitHub