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.