Skip to content

bafu/neural-network-model-manipulations

Repository files navigation

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

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published