Neural Network Toolbox on TensorFlow
Project description
Neural Network Toolbox on TensorFlow.
See some examples to learn about the framework:
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:
The 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. Not focus on models. + There are already too many symbolic function wrappers. Tensorpack includes only a few common models, and helpful tools such as LinearWrap to simplify large models. But you can use any other wrappers within tensorpack, such as sonnet/Keras/slim/tflearn/tensorlayer/….
Focus on training speed.
Tensorpack trainer is almost always faster than feed_dict based wrappers. Even on a tiny CNN example, the training runs 2x faster than the equivalent Keras code.
Data-parallel multi-GPU training is off-the-shelf to use. It is as fast as Google’s benchmark code.
Data-parallel distributed training is off-the-shelf to use. It is as slow as Google’s benchmark code.
Focus on large datasets.
It’s painful to read/preprocess data from TF. Use DataFlow to load large datasets (e.g. ImageNet) in pure Python with multi-process prefetch.
DataFlow has a unified interface, so you can compose and reuse them to perform complex preprocessing.
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
Run inference on a test dataset
Run some operations once a while
Send loss to your phone
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.
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.3.0-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 317fc3d499f2ffbc81b93197a0a3d1d8cec284865b8595d28d0949b38f7771ad |
|
MD5 | 8727af99d4c5b6810ab46d15a1fcd5b1 |
|
BLAKE2b-256 | e9b647d4d1af7164977f926815922754a2b6db9c4ea04a36a53e4d24d74e0001 |