Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Mario authored Jan 21, 2018
1 parent 417e89c commit 93076a8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ Pruing is one of the hot topics on reduction of computing resources requiered by
## Dependencies

The base of this code was written and shared by [He Yihui](https://github.com/yihui-he), and runs on his costum branch of CAFFE (python 3 implementation).
After cloning the repo, build the CAFFE binaries by following the instructions on the brach [repo](https://github.com/yihui-he/caffe-pro).
After cloning the repo, build the CAFFE binaries by following the instructions in the branch [repo](https://github.com/yihui-he/caffe-pro).
Train or donwload a caffemodel and place it in the temp directory.
Make sure to modify cfgs.py with the layer names and pruning degree (i.e. number of channels that will be retained after the pruning process) you desire for each layer.

Expand Down

0 comments on commit 93076a8

Please sign in to comment.