Merge pull request BVLC#5521 from kkhoot/fix_lstm_unit_check

remove redundant check in LSTMUnitLayer
haikuoyao · Apr 13, 2017 · f522d9c · f522d9c
2 parents 3d5bed0 + 3a98796
commit f522d9c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/caffe/layers/lstm_unit_layer.cpp b/src/caffe/layers/lstm_unit_layer.cpp
@@ -31,7 +31,6 @@ void LSTMUnitLayer<Dtype>::Reshape(const vector<Blob<Dtype>*>& bottom,
     CHECK_EQ(num_instances, bottom[i]->shape(1));
   }
   hidden_dim_ = bottom[0]->shape(2);
-  CHECK_EQ(num_instances, bottom[1]->shape(1));
   CHECK_EQ(4 * hidden_dim_, bottom[1]->shape(2));
   top[0]->ReshapeLike(*bottom[0]);
   top[1]->ReshapeLike(*bottom[0]);