Closed dlyldxwl closed 6 years ago
caffe adds new layer via cpp file,in the file of depthwise_conv_layer.cpp,you can find REGISTER_LAYER_CLASS(DepthwiseConvolution);,Only when you want to add new parameters for this layer, these parameters can be added in caffe.proto.the depthwiseconvlution use the convolution_param layer { name: "conv2_1/dw" type: "DepthwiseConvolution" bottom: "conv1" top: "conv2_1/dw" param { lr_mult: 1.0 decay_mult: 1.0 } convolution_param { num_output: 32 bias_term: false pad: 1 kernel_size: 3 group: 32 stride: 1 weight_filler { type: "msra" } } } and This convolution_param already exists in caffe.proto, so there is no need to add new layer_param
i think you picture is large,so the speed is slow
@mameng1 ok, i get it, thank you. can you tell me the role of cu file? and when to use cu file? thanks!
if you Compile the cu file,and caffe.set_mode_gpu(),It will automatically call the cu file
@mameng1 Thanks again!
Thanks for your job! But I have some questions to ask you , I hope you can help me.