Skip to content

Commit 82e6911

Browse files
committed
change recur_name_prefix into recur_input_names
1 parent 18aa786 commit 82e6911

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/caffe/layers/rnn_v2_layer.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -552,9 +552,9 @@ void RNNv2Layer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
552552
// Setup pointers to paired recurrent inputs/outputs.
553553
recur_input_blobs_.resize(num_recur_blobs);
554554
recur_output_blobs_.resize(num_recur_blobs);
555-
for (int i = 0; i < recur_name_prefix.size(); ++i) {
555+
for (int i = 0; i < recur_input_names.size(); ++i) {
556556
recur_input_blobs_[i] =
557-
CHECK_NOTNULL(unrolled_net_->blob_by_name(recur_name_prefix[i]).get());
557+
CHECK_NOTNULL(unrolled_net_->blob_by_name(recur_input_names[i]).get());
558558
recur_output_blobs_[i] =
559559
CHECK_NOTNULL(unrolled_net_->blob_by_name(recur_output_names[i]).get());
560560
}

0 commit comments

Comments
 (0)