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
|
||||
|
||||
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
|
||||
```
|
||||
Reference in New Issue
Block a user