提交 bd3f976a 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #906 from emailweixu/check_cnn_input_size

Check the input size for convolution.
...@@ -130,6 +130,10 @@ void ConvProjection::reshapeTensorDesc(int batchSize) { ...@@ -130,6 +130,10 @@ void ConvProjection::reshapeTensorDesc(int batchSize) {
void ConvProjection::reshape(int batchSize) { void ConvProjection::reshape(int batchSize) {
size_t width = calOutputSize(); size_t width = calOutputSize();
CHECK_EQ(width, out_->value->getWidth()); 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_); isSelectAlgo_ = (batchSize == batchNum_);
batchNum_ = batchSize; batchNum_ = batchSize;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册