fix bug in reverse (backward) direction of RNNv2

haifenghan · haifenghan · commit 3d3f6b1744c4 · 2020-09-28T21:23:22.000+08:00
diff --git a/src/caffe/layers/rnn_v2_layer.cpp b/src/caffe/layers/rnn_v2_layer.cpp
@@ -405,7 +405,7 @@ void RNNv2Layer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
       }
     }
     //
-    this->FillUnrolledNet(&net_param, "x_rev", "cont", output_names, recur_name_prefix, "bw_");
+    this->FillUnrolledNet(&net_param, "x_rev", "cont_rev", output_names, recur_name_prefix, "bw_");
     {
       // reverse output back
       LayerParameter *reverse_output_layer_param = net_param.add_layer();

Original file line number	Diff line number	Diff line change
`@@ -405,7 +405,7 @@ void RNNv2Layer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,`
`405`	`405`	`}`
`406`	`406`	`}`
`407`	`407`	`//`
`408`		`- this->FillUnrolledNet(&net_param, "x_rev", "cont", output_names, recur_name_prefix, "bw_");`
	`408`	`+ this->FillUnrolledNet(&net_param, "x_rev", "cont_rev", output_names, recur_name_prefix, "bw_");`
`409`	`409`	`{`
`410`	`410`	`// reverse output back`
`411`	`411`	`LayerParameter *reverse_output_layer_param = net_param.add_layer();`