cuda_nn_mnist
Neural network library for digit classification powered by cuda
Resources
The library was built to work with MNIST dataset.
python-mnist data parser was used to load the dataset.
Hardware requirements
The library uses Numba which
supports
CUDA-enabled GPUs with compute capability 2.0 or above with an up-to-date Nvidia driver.
See the list of CUDA-enabled GPU cards
Trained weights
Trained weights are located in weights
directory. Call load_weights
method to use them.