A fork from Marvin, and simplify this project for learning Deep Learning.
This is a neural network framework using c++11 and CUDA.
Download CUDA 7.5 and [cuDNN 5.1](https://developer.nvidia.com/ cudnn).
Config you cuda path in compile.sh.
sh compile.sh
Convolutional Neural Network on MNIST digits.
-
Prepare data: run examples/mnist/prepare_mnist.m in Matlab
-
Config your neural network model in /examples/mnist,just like this file lenet.json.
-
Visualize filters: config your json file path in /tools/viewer_matlab/plotNet.m,and run it in Matlab. See you train result like this:
-
Train a model: run ./examples/mnist/train.sh in shell. See you train result like this:
-
Visualize different model change the train result: use /tools/picture.py. Before run it,you need config:
file_list:the different models tained result output in shell,and save as files.
dirname:the dir of file_list.