您的位置:首页 > Web前端

[Caffe]:关于Check failed: registry.count(type) == 0 (1 vs. 0)错误

2017-05-22 10:09 501 查看

错误描述

当我在终端下用caffe命令对带有PythonLayer的网络训练时,出现了如下错误。



该错误提示Layer的卷积类型已经注册过了。为什么会出现这个错误呢?

其实是,博主机子上有多个Caffe版本,该版本的pycaffe接口没有被指定正确。

解决方法

我的一种方法是在train.sh脚本文件开头,添加指定PYTHONPATH。

即,在train.sh文件中

#train.sh
export PYTHONPATH=/path/to/your/caffe/python/:$PYTHONPATH #把caffe/python路径加在PYTHONPATH最前面,优先检索
TOOLS=/path/to/your/caffe/build/tools #指定命令路径,即指定caffe版本
$TOOLS/caffe train --solver="" blablabla...


加完路径后一切就恢复正常啦。

2017-5-22 记。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐