mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
Update README
This commit is contained in:
41
README.md
41
README.md
@@ -1,5 +1,40 @@
|
|||||||
# CUDANet
|
# CUDANet
|
||||||
|
|
||||||
requirements:
|
:warning: Work in progress
|
||||||
- CUDA
|
|
||||||
- Google Test
|
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
|
||||||
|
```
|
||||||
Reference in New Issue
Block a user