bafu / neural-network-model-manipulations

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Introduction

Simple examples demonstrating techniques of deep learning model manipulation.

  • mnist-nn.py: 3-layer neural network as digit recognizer (MNIST).
  • mnist-nn-gpu.py: mnist-nn with GPU computing.
  • mnist-nn-data-parallelism.py: mnist-nn with data parallelism (MPI and GPU).

Setup development environment

Please refer to .

Misc.

src/dataset/* are dataset retrivers copied from Keras.

About


Languages

Language:Jupyter Notebook 80.5%Language:Python 19.0%Language:Shell 0.5%