CUDANet/README.md

CUDANet

⚠️ Work in progress

Convolutional Neural Network inference library running on CUDA.

Features

• Input layer
• Dense (fully-connected) layer
• Conv2d layer
• Max pooling
• Average pooling
• Concat layer
• Sigmoid activation
• ReLU activation
• Softmax activation
• Load weights from file

Usage

requirements

• cmake
• CUDA
• Google Test (for testing only)

build

mkdir build
cd build
cmake -S ..
make

build and run tests

make test_main
./test/test_main