Neural Network Toolbox on TensorFlow
Project description
A neural net training interface based on TensorFlow.
See some examples to learn about the framework. Everything runs on multiple GPUs, because why not?
Vision:
Generative Adversarial Network(GAN) variants, including DCGAN, InfoGAN, Conditional GAN, WGAN, BEGAN, DiscoGAN, Image to Image, CycleGAN.
Fully-convolutional Network for Holistically-Nested Edge Detection(HED)
Reinforcement Learning:
Deep Q-Network(DQN) variants on Atari games, including DQN, DoubleDQN, DuelingDQN.
Asynchronous Advantage Actor-Critic(A3C) with demos on OpenAI Gym
Speech / NLP:
Examples are not only for demonstration of the framework – you can train them and reproduce the results in papers.
Features:
It’s Yet Another TF wrapper, but different in: 1. It’s not a model wrapper. + There are already too many symbolic function wrappers. Tensorpack includes only a few common models, but you can use any other model wrappers within tensorpack, such as sonnet/Keras/slim/tflearn/tensorlayer/….
Focus on training speed.
Speed comes for free with tensorpack – it uses TensorFlow in the correct way. On various CNNs, it runs 1.5~1.7x faster than the equivalent Keras code.
Data-parallel multi-GPU/distributed training is off-the-shelf to use. It is as fast as Google’s official benchmark.
See tensorpack/benchmarks for some benchmark scripts.
Focus on large datasets.
It’s painful to read/preprocess data through TF. tensorpack helps you load large datasets (e.g. ImageNet) in pure Python with autoparallelization. It also naturally works with TF Queues or tf.data.
Interface of extensible Callbacks. Write a callback to implement everything you want to do apart from the training iterations, and enable it with one line of code. Common examples include:
Change hyperparameters during training
Print some tensors of interest
Monitor GPU utilization
Send error rate to your phone
See tutorials to know more about these features.
Install:
Dependencies:
Python 2 or 3
TensorFlow >= 1.0.0 (>=1.1.0 for Multi-GPU)
Python bindings for OpenCV (Optional, but required by a lot of features)
pip install -U git+https://github.com/ppwwyyxx/tensorpack.git # or add `--user` to avoid system-wide installation.
If you only want to use tensorpack.dataflow alone as a data processing library, TensorFlow is also optional.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for tensorpack-0.6.1-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9f567639ce206579a26180d5e6b454363ec93a5ff138c9470df12bb071161ab4 |
|
MD5 | 184d8d3afd4f11cce95ea9c2ee4bfc3a |
|
BLAKE2b-256 | 398daceaa143a487cdcf998e3287c989a52d6514e0e0f5f97249b4096e8f4bc0 |