Update CudnnConvLayer.cpp

avx_docs
qingqing01 8 years ago committed by GitHub
parent 95da095d10
commit c1c07bbef6

@ -133,6 +133,8 @@ void CudnnConvLayer::reshape(int batchSize) {
getOutput().setFrameHeight(outputH_);
getOutput().setFrameWidth(outputW_);
// if the batchSize remains the same, set isSelectAlgo_ true.
// Otherwise, set isSelectAlgo_ false and select algo again.
isSelectAlgo_ = (batchSize == batchNum_);
batchNum_ = batchSize;

Loading…
Cancel
Save