在PyCaffe中定义的层模块在哪里
在PyCaffe中定义的层模块在哪里
提问于 2016-03-23 20:11:43
我正在修改一个咖啡教程以实现一个神经网络，但我很难确定某些pycaffe模块的位置，以便查看某些函数定义。
例如，本教程提到：
import caffe
from caffe import layers a L, params as P
....
L.Convolution(bottom, kernel_size=ks, stride=stride, num_output=nout, pad=pad, group=group)
L.InnerProduct(bottom, num_output=nout)
L.ReLU(fc, in_place=True)
...在哪里可以找到这些函数定义，在哪里可以看到哪些其他类型的层是预定义的？我看到layers和params是在此定义，但是没有提到类型(例如layers.Convolution等)。
我之所以试图找出这一点，是因为在pycaffe教程中遗漏了其他prototxt参数，在生成原型时，我希望能够从Python中定义这些参数。其中包括，blob_lr和include{phase: TRAIN}。
回答 1
Stack Overflow用户
回答已采纳
发布于 2016-03-27 07:35:04
您可以像这样添加blob_lr和phase：
import caffe
from caffe import layers a L, params as P
ns = caffe.NetSpec()
ns.conv = L.Convolution(bottom, convolution_param={'kernel_size':ks,
'stride':stride,
'num_output':nout,
'pad':pad,
'group':group},
param=[{'lr_mult':1, 'decay_mult':1},
{'lr_mult':2, 'decay_mult':0}],
include={'phase': caffe.TRAIN})您可以在这个答案中看到更多的示例。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/36187825
复制相关文章
相似问题
腾讯云开发者
