Check the input size for convolution.

Make sure the input size is correct. Change-Id: I2559075e32a098c5ed51942ff8351d964a6d6c75
10 years ago · fa33262d2e
parent b7157d4485
commit fa33262d2e
1 changed files with 6 additions and 0 deletions
--- a/paddle/gserver/layers/ConvProjection.cpp
+++ b/paddle/gserver/layers/ConvProjection.cpp
@ -130,6 +130,12 @@ void ConvProjection::reshapeTensorDesc(int batchSize) {
 void ConvProjection::reshape(int batchSize) {
  size_t width = calOutputSize();
  CHECK_EQ(width, out_->value->getWidth());
+  CHECK_EQ(channels_ * imageH_ * imageW_, in_->value->getWidth())
+          << "Wrong input size for convolution"
+          << " channels=" << channels_
+          << " imageH=" << imageH_
+          << " imageW=" << imageW_
+          << " inputSize=" << in_->value->getWidth();

  isSelectAlgo_ = (batchSize == batchNum_);
  batchNum_ = batchSize;