bityangke / tinynn Goto Github PK

tinynn is a lightweight deep learning framework written in Python3 (with NumPy).

pip install tinynn

git clone https://github.com/borgwang/tinynn.git
cd tinynn/examples

# MNIST classification
python mnist/run.py  

# a toy regression task
python nn_paint/run.py  

# reinforcement learning demo (gym environment required)
python rl/run.py

• layers: Dense, Conv2D, ConvTranspose2D, RNN, MaxPool2D, Dropout, BatchNormalization
• activation: ReLU, LeakyReLU, Sigmoid, Tanh, Softplus
• losses: SoftmaxCrossEntropy, SigmoidCrossEntropy, MAE, MSE, Huber
• optimizer: RAdam, Adam, SGD, Momentum, RMSProp, Adagrad, Adadelta

Please follow the Google Python Style Guide for Python coding style.

In addition, please sort the module import order alphabetically in each file. To do this, one can use tools like isort (be sure to use --force-single-line-imports option to enforce the coding style).

MIT