首页 > 技术文章 > 添加新网络模型后运行报错:未定义的参数:ps_roipooling param

rainsoul 2019-03-11 18:29 原文

现象描述:在新增了具有自定义的data层或者loss层的网路之后,工程运行会报错;

疑惑:并没有这样的参数新增,并且前向的deploy文件已经将自定义的loss以及data等都去掉了;

可能的原因:虽然在deploy文件中去掉了,但是在caffemodel中还是存在着的,在解析的时候不认识自定义层,导致无法解析成deploy文件中的参数,随便报了一个错;

解决方法:将自定义的data以及loss层等用脚本抽取掉,再使用就没有问题了;

推荐阅读