首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Caffe中层参数的动态修改

Caffe中层参数的动态修改
EN

Stack Overflow用户
提问于 2016-11-29 05:40:35
回答 3查看 1K关注 0票数 1

我使用折叠代码加载和设置网络,层的参数存储在deploy.prototxt中。

net = caffe.Net(deploy.prototxt, caffemodel, caffe.TEST)

然而,我想要做的是修改参数(例如kernel_sizepad等)。而不是修改prototxt文件并重新加载它。有没有办法这样做?

EN

Stack Overflow用户

回答已采纳

发布于 2017-04-06 20:36:11

您可以编写自己的get/set方法并将它们公开给python。在layer.hpp中:

代码语言:javascript
运行
复制
virtual float GetParameter(const std::string param_name) {return -1;}
virtual void SetParameter(const std::string param_name, float val) {}

然后在希望动态获取/设置参数的层中重新定义这些方法。

最后一步是向python公开这些方法。在_caffe.cpp中为bp::class_<Layer...添加以下内容

代码语言:javascript
运行
复制
.def("get_parameter", &Layer<Dtype>::GetParameter)
.def("set_parameter", &Layer<Dtype>::SetParameter)
票数 0
EN
查看全部 3 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/40858548

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档