如何使用caffe进行int8和tensorrt训练

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/csuzhaoqinghui/article/details/79990953

首先,在打patch之前需要做两件事,一个是确认未打patch之前训练测试是没有问题的,二最好先做好备份,以防止打patch过程中出现问题无法修复。
以下是几个重要步骤:
1. cd到caffe根目录下,执行命令make clean,将之前编译好的build等文件删除
2. 用git am $PATH/fpga.8-bit..patch命令尝试是否能进行打patch.
3. 如果不能通过上一个命令打,报这个错误:Patch format detection failed,那直接用patch命令:’patch -p1 < fpga.8-bit.train.2017-11-07.patch’
4. 如果得到以下输入,则说明大部分成功了, 有两处失败;(图片在youdaoimage笔记里)
5. 需要手动改一下 src/caffe/proto/caffe.proto ,在message LayerParameter { }的最后加上如下3行. // FPGA Quantization Parameter optional QuantizationParameter quantization_param = 6000; // add by liyupeng03 手动把src/caffe/proto/caffe.proto.rej 里的东西加到caffe.proto
6. 把Makefile.config.rej里的东西加到Makefile.config中
7. 然后切到caffe的根目录下,重新make all编译
8. patch包地址:

猜你喜欢

转载自blog.csdn.net/csuzhaoqinghui/article/details/79990953