|
|
|
@ -131,11 +131,9 @@ 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();
|
|
|
|
|
<< "Wrong input size for convolution"
|
|
|
|
|
<< " channels=" << channels_ << " imageH=" << imageH_
|
|
|
|
|
<< " imageW=" << imageW_ << " inputSize=" << in_->value->getWidth();
|
|
|
|
|
|
|
|
|
|
isSelectAlgo_ = (batchSize == batchNum_);
|
|
|
|
|
batchNum_ = batchSize;
|
|
|
|
|