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

Update CudnnConvLayer.cpp

上级 95da095d
...@@ -133,6 +133,8 @@ void CudnnConvLayer::reshape(int batchSize) { ...@@ -133,6 +133,8 @@ void CudnnConvLayer::reshape(int batchSize) {
getOutput().setFrameHeight(outputH_); getOutput().setFrameHeight(outputH_);
getOutput().setFrameWidth(outputW_); getOutput().setFrameWidth(outputW_);
// if the batchSize remains the same, set isSelectAlgo_ true.
// Otherwise, set isSelectAlgo_ false and select algo again.
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.
先完成此消息的编辑!
想要评论请 注册