Commit a4867550 authored by Yuxin Wu's avatar Yuxin Wu

update readme

parent f74f8900
......@@ -2,7 +2,7 @@ Code and model for the paper:
[DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradients](http://arxiv.org/abs/1606.06160), by Zhou et al.
We hosted a demo at CVPR16 on behalf of Megvii, Inc, running real-time 1/4 VGG size DoReFa-Net on both ARM and FPGA.
We hosted a demo at CVPR16 on behalf of Megvii, Inc, running a real-time 1/4-VGG size DoReFa-Net on ARM and half-VGG size DoReFa-Net on FPGA.
We're not planning to release those runtime bit-op libraries for now. In these examples, bit operations are run in float32.
Pretrained model for 1-2-6-AlexNet is available at
......
......@@ -127,6 +127,7 @@ class MultiThreadAsyncPredictor(AsyncPredictorBase):
"""
def __init__(self, predictors, batch_size=5):
""" :param predictors: a list of OnlinePredictor"""
assert len(predictors)
for k in predictors:
#assert isinstance(k, OnlinePredictor), type(k)
# TODO use predictors.return_input here
......
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