tinynn
tinynn is a lightweight deep learning framework written in Python3 (with NumPy).
Getting Started
Install
pip install tinynn
Examples
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
Components
- 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