Skip to content

Latest commit

 

History

History
16 lines (12 loc) · 761 Bytes

README.md

File metadata and controls

16 lines (12 loc) · 761 Bytes

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.