Commit db5019b8 authored by Yuxin Wu's avatar Yuxin Wu

fix build

parent fb4dd633
......@@ -4,7 +4,7 @@ Neural Network Toolbox on TensorFlow
[![Build Status](https://travis-ci.org/ppwwyyxx/tensorpack.svg?branch=master)](https://travis-ci.org/ppwwyyxx/tensorpack)
[![badge](https://readthedocs.org/projects/pip/badge/?version=latest)](http://tensorpack.readthedocs.io/en/latest/index.html)
Tutorials are not finished. See some [examples](examples) to learn about the framework:
See some [examples](examples) to learn about the framework:
### Vision:
+ [DoReFa-Net: train binary / low-bitwidth CNN on ImageNet](examples/DoReFa-Net)
......@@ -53,7 +53,7 @@ With the above components defined, tensorpack trainer runs the training iteratio
Even on a small CNN example, the training runs [2x faster](https://gist.github.com/ppwwyyxx/8d95da79f8d97036a7d67c2416c851b6) than the equivalent Keras code.
Multi-GPU training is off-the-shelf by simply switching the trainer.
You can also define your own trainer for non-standard training (e.g. GAN).
You can also define your own trainer for different style of training (e.g. GAN).
## Install:
......
......@@ -350,8 +350,8 @@ from ..utils.develop import create_dummy_func # noqa
try:
import matplotlib.pyplot as plt
except ImportError:
pyplot2img = create_dummy_func('pyplot2img') # noqa
intensity_to_rgb = create_dummy_func('intensity_to_rgb') # noqa
pyplot2img = create_dummy_func('pyplot2img', 'matplotlib') # noqa
intensity_to_rgb = create_dummy_func('intensity_to_rgb', 'matplotlib') # noqa
if __name__ == '__main__':
if False:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment