Commit 87569ca6 authored by Yuxin Wu's avatar Yuxin Wu

update dorefa doc

parent bbe8f42a
...@@ -15,7 +15,8 @@ See some interesting [examples](examples) to learn about the framework: ...@@ -15,7 +15,8 @@ See some interesting [examples](examples) to learn about the framework:
Abstract your training task into three components: Abstract your training task into three components:
1. Model, or graph. `models/` has some scoped abstraction of common models. 1. Model, or graph. `models/` has some scoped abstraction of common models. This part is roughly an
equivalent of slim/tflearn/tensorlayer.
`LinearWrap` and `argscope` makes large models look simpler. `LinearWrap` and `argscope` makes large models look simpler.
2. Data. tensorpack allows and encourages complex data processing. 2. Data. tensorpack allows and encourages complex data processing.
...@@ -32,7 +33,7 @@ Abstract your training task into three components: ...@@ -32,7 +33,7 @@ Abstract your training task into three components:
+ Send the accuracy to your phone + Send the accuracy to your phone
With the above components defined, tensorpack trainer will run the training iterations for you. With the above components defined, tensorpack trainer will run the training iterations for you.
Multi-GPU training is ready to use by simply switching the trainer. Multi-GPU training is off-the-shelf by simply switching the trainer.
## Dependencies: ## Dependencies:
......
...@@ -289,8 +289,7 @@ if __name__ == '__main__': ...@@ -289,8 +289,7 @@ if __name__ == '__main__':
parser.add_argument('--load', help='load a checkpoint, or a npy (given as the pretrained model)') parser.add_argument('--load', help='load a checkpoint, or a npy (given as the pretrained model)')
parser.add_argument('--data', help='ILSVRC dataset dir') parser.add_argument('--data', help='ILSVRC dataset dir')
parser.add_argument('--dorefa', parser.add_argument('--dorefa',
help='number of bits for W,A,G, separated by comma. Defaults to \'1,2,4\'', help='number of bits for W,A,G, separated by comma', required=True)
default='1,2,4')
parser.add_argument('--run', help='run on a list of images with the pretrained model', nargs='*') parser.add_argument('--run', help='run on a list of images with the pretrained model', nargs='*')
args = parser.parse_args() args = parser.parse_args()
......
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