caffe研究与应用
seasermy
专注AI算法研究与应用,AI软硬件协同优化,AI系统解决方案
展开
-
如何将caffe的data prefetch变为多线程预取?
在深度学习研究时,数据预取会经常成为训练的瓶颈,一般来说,判断瓶颈的时候,看GPU利用率即可。通过nvidia-smi或者watch -n 1 nvidia-smi(实时刷新)都可查看GPU的利用率。 如上图所示,GPU利用率不能完全打满,断断续续的忽高忽低,大概率是因为数据取出速度慢 (其他原因,如GPU带宽一般不会成为瓶颈)。除此之外,在blocking_queue.cpp中也会打印出...原创 2019-04-17 10:31:54 · 1583 阅读 · 1 评论 -
windows下用caffe加载二进制模型(linux下训练)的问题
最近,需要移植faster-rcnn的detect部分到android平台上,为方便删减代码与调试,需要跨平台兼容到windows下运行,windows下调试的时候,用的是linux下的模型定义proto与训练好的二进制模型,但是,一直加载模型不成功,逐步解决方法如下: (1)核对protobuf版本,均为2.5.0,不可能是因为版本不兼容所致; (2)核对cafe.proto,这个文件...原创 2016-05-26 18:55:20 · 3108 阅读 · 5 评论 -
如何关闭caffe库中的LOG输出
在编译libcaffe.so的时候,在src/caffe/net.cpp Init函数进入的时候加入 fLI::FLAGS_minloglevel=3; 即可原创 2019-08-29 10:37:19 · 498 阅读 · 0 评论 -
caffe中thread-safe问题解决
caffe并不是thread safe的,在使用深度学习框架部署的实践中,一定注意这个问题, 其他的框架如mxnet, tensorflow也是如此,那么如何解决caffe的这个问题?先明确一个线程的概念: 1. boost::thread_specific_ptr 什么是线程不安全呢?多个线程执行同一段代码或接口,如果代码或接口中含有共享的变量,那么会引起线程竞争,引发conflict...原创 2019-09-02 14:39:18 · 348 阅读 · 0 评论