caffe 添加crnn所需的层

lijingle 深度学习框架 2022-10-9 09:22 1724人围观

本文介绍caffe在编译crnn中添加的层时所遇到的问题。

1.添加continuation_indicator_layer.cpp 和continuation_indicator_layer.cu文件以及continuation_indicator_layer.h文件

2.添加ctc_loss_layer.cpp 和ctc_loss_layer.h文件以及相关的3rdparty文件夹

在caffe.proto里面添加要添加的层。

  optional CtcLossParameter ctc_loss_param = 200;
  optional ContinuationIndicatorParameter continuation_indicator_param = 201;

message CtcLossParameter {
    optional uint32 alphabet_size = 1 [default = 0];
    optional uint32 time_step = 3 [default = 0];
    optional int32 blank_label = 4 [default = 0];
}
message ContinuationIndicatorParameter {
    optional uint32 time_step = 1 [default = 0];
    optional uint32 batch_size = 2 [default = 0];
}

在编译caffe中会出现cudnn舍弃的一些接口，这里只需将其接口进行修改即可

src/reduce.cu
Line 44 to : shuff = __shfl_down_sync(0xFFFFFFFF, x, offset);

include/contrib/moderngpu/include/device/intrinsics.cuh
Line 115 to : var = __shfl_up_sync(0xFFFFFFFF, var, delta, width);
Line 125 to : p.x = __shfl_up_sync(0xFFFFFFFF, p.x, delta, width);
Line 126 to : p.y = __shfl_up_sync(0xFFFFFFFF, p.y, delta, width);
Line 143 to : "shfl.up.sync.b32 r0|p, %1, %2, %3, %4;"
Line 158 to : "shfl.up.sync.b32 r0|p, %1, %2, %3, %4;"