Update README

This commit is contained in:
2024-03-17 16:18:42 +01:00
parent 0c22fac64e
commit b1621819ca

View File

@@ -1,5 +1,40 @@
# CUDANet
requirements:
- CUDA
- Google Test
:warning: Work in progress
Convolutional Neural Network inference library running on CUDA.
## Features
- [x] Input layer
- [x] Dense (fully-connected) layer
- [x] Conv2d layer
- [ ] Max pooling
- [ ] Average pooling
- [ ] Concat layer
- [x] Sigmoid activation
- [x] ReLU activation
- [ ] Softmax activation
- [ ] Load weights from file
## Usage
**requirements**
- [cmake](https://cmake.org/)
- [CUDA](https://developer.nvidia.com/cuda-downloads)
- [Google Test](https://github.com/google/googletest) (for testing only)
**build and test**
```sh
mkdir build
cd build
cmake -S ..
make
```
Run tests
```sh
./test/test_main
```