![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
caffe
国服卤蛋儿
这个作者很懒,什么都没留下…
展开
-
Check failed: status == CUBLAS_STATUS_SUCCESS (11 vs. 0) CUBLAS_STATUS_MAPPING_ERROR
用KolektorSDD数据集跑caffe的unet没什么问题,虽然结果检测不到缺陷 遂用C1数据集尝试。但是训练时候报了如题的错误。 有的说是index越界,貌似没有。 别人的说法: 1.Check failed: status == CUBLAS_STATUS_SUCCESS (11 vs. 0) CUBLAS_STATUS_MAPPING_ERROR - thesby的专栏 https://...原创 2019-12-23 18:13:56 · 683 阅读 · 0 评论 -
使用caffe的python接口裁剪caffemodel
最近有个需求就是将训练好的caffemodel裁剪保留前几层,然乎保存为新的caffemodel。 找了一些资料都是解析或修改caffemodel参数,和结构相关的只有把caffemodel的最后的fc层通过flat转化成卷积层。 找到相关的,只不过是用caffe的matlab接口实现的。还是很有参考价值。由于不太好找,所以在这里记录下链接。 具体方法:copy已有的deploy文件保存为depl...原创 2019-11-29 11:05:41 · 294 阅读 · 0 评论