Skip to content

Commit

Permalink
-readme change
Browse files Browse the repository at this point in the history
  • Loading branch information
Tyill committed Jul 29, 2019
1 parent 08a41e7 commit 8362d56
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 9 deletions.
12 changes: 11 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,16 @@ SkyNet is a light deep learning library.
[ResNet cpp-example for Win](https://github.com/Tyill/storage/tree/master/resnetDemo/Builds)


Compare with Tensorflow, inference ResNet50. [Code.] (https://github.com/Tyill/skynet/blob/master/example/resnet50/compareWithTF.py)
PC: i5-2400, GF1050, Win7, MSVC12.

| | **CPU: time/img, ms** | **GPU: time/img, ms** | **CPU: RAM, Mb** | **GPU: RAM, Mb** |
|------------------|-----------------------|-----------------------|------------------|------------------|
| Skynet | 195 | 15 | 600 | 800 |
|--------------------------------------------------------------------------------------------------------|
| Tensorflow | 250 | 25 | 400 | 1400 |


## Features

* the library is written from scratch in C++ (only STL + OpenBLAS for calculation), C-interface
Expand Down Expand Up @@ -68,7 +78,7 @@ for n in range(1000):

* pip install libskynet - CPU

* pip install libskynet-cu - CPU + cuDNN7.3.1
* pip install libskynet-cu - CUDA + cuDNN7.3.1


## [Wiki](https://github.com/Tyill/skynet/wiki)
Expand Down
7 changes: 3 additions & 4 deletions example/mnist/cpp_example.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -122,14 +122,13 @@ int main(int argc, char* argv[]){

// training
float accurat = 0;
/* snet.training(lr,
snet.training(lr,
inLayer,
outLayer,
targetLayer,
accurat);*/

snet.forward(false, inLayer, outLayer);
accurat);


// calc error
sn::snFloat* targetData = targetLayer.data();
sn::snFloat* outData = outLayer.data();
Expand Down
8 changes: 4 additions & 4 deletions python/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,15 +37,15 @@
with open("README.md", "r") as fh:
long_description = fh.read()

__version__ = '1.0.3'
__version__ = '1.0.4'

setup(
name="libskynet",
name="libskynet-cu",
version=__version__,
url="https://github.com/Tyill/skynet",
packages=find_packages(),
package_dir={'libskynet': 'libskynet'},
package_data={'libskynet': ['*.so', '*.dll']},
package_dir={'libskynet-cu': 'libskynet-cu'},
package_data={'libskynet-cu': ['*.so', '*.dll']},
description="neural net with blackjack and hookers",
long_description=long_description,
long_description_content_type="text/markdown",
Expand Down

0 comments on commit 8362d56

Please sign in to comment.