Update CudnnConvLayer.cpp

vincemo · Sep 23, 2016 · c1c07bb · c1c07bb
1 parent 95da095
commit c1c07bb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paddle/gserver/layers/CudnnConvLayer.cpp b/paddle/gserver/layers/CudnnConvLayer.cpp
@@ -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;