This CNN is written by pure numpy and support trainning by BP algorithem.
- It follows pytorch's code style(forward, backward).
- Due to numpy limiation, it doesn't use GPU to accelerate. But the training speed is acceptable for the small dataset,(e.g. mnist).
- It implements the SGD (Stochastic Gradient Descent)
- It support multiple channel.
Downlaod the mnist from website, and extract all the files into data/mnist
python mnist_cnn.py