Skip to content

Paseam/tensorpack

 
 

Repository files navigation

tensorpack

Neural Network Toolbox on TensorFlow

Still in development, but usable.

See some interesting examples to learn about the framework:

Features:

Focused on modularity. Just have to define the three components in training:

  1. The model, or the graph. Define the graph as well as its inputs and outputs. models/ has some scoped abstraction of common models.

  2. The data. All data producer has an unified DataFlow interface, and this interface can be chained to perform complex preprocessing. It uses multiprocess to avoid performance bottleneck on data loading.

  3. The callbacks. They include everything you want to do apart from the training iterations: change hyperparameters, save models, print logs, run validation, and more.

With the above components defined, tensorpack trainer will run the training iterations for you. Multi-GPU training is ready to use by simply changing the trainer.

Dependencies:

  • Python 2 or 3
  • TensorFlow >= 0.8
  • Python bindings for OpenCV
  • other requirements:
pip install --user -r requirements.txt
pip install --user -r opt-requirements.txt (some optional dependencies, you can install later if needed)
  • Use tcmalloc whenever possible: see TF issue
  • allow import tensorpack everywhere:
export PYTHONPATH=$PYTHONPATH:`readlink -f path/to/tensorpack`

About

Neural Network Toolbox on TensorFlow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%