tennisgazelle/cuda-capsulenetwork-methods

A clean, pure C++/CUDA implementation of Capsule Networks, no cuDNN, TF, Keras, or libraries.

Neural Nets

By Daniel Lopez Computer Science Graduate Student at the University of Nevada, Reno.

Three implementations of Neural Nets, built on top of one another for CUDA implementation and further researach/development. Written in C++/CUDA for full control ove data management and moving (not implying TensorFlow or other projects are badly written, or don't allow low-level control of data)

Building/Running

The preferred IDE is CLion by JetBrains, but choose whatever floats your boat :)

Dependencies

This project is build with CMake and only uses the armadillo library as dependency. Instructions for Armadillo download installation may be found on their website.

Building

mkdir build
cd build
cmake ..

Running

cd build/
./NeuralNets

Nothing too special comes up yet, look at the file src/main.cpp to see what you're supposed to do right now.