tinynn is a lightweight deep learning framework written in Python3 (with NumPy).
There are two branches (master
and mini
) in thie repo. The mini
branch contains the minimal components to run a neural network. The master
branch holds the latest stable code with more components and features. See Components down below.
git clone https://github.com/borgwang/tinynn.git
cd tinynn
pip install -r requirements.txt
cd tinynn
# MNIST classification
python examples/mnist/run.py
# a toy regression task
python examples/nn_paint/run.py
# reinforcement learning demo (gym environment required)
python examples/rl/run.py
- layers: Dense, Conv2D, MaxPool2D, Dropout
- activation: ReLU, LeakyReLU, Sigmoid, Tanh, Softplus
- losses: SoftmaxCrossEntropy, SigmoidCrossEntropy, MAE, MSE, Huber
- optimizer: SGD, Adam, 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