Merge pull request #999 from qingqing01/bug_fix

Bug fix for BatchNorm when testing more than models in test mode.

Author: lcy-seso

paddle/gserver/layers/BatchNormalizationLayer.h

@@ -58,6 +58,8 @@ class BatchNormalizationLayer : public BatchNormBaseLayer {
   /// to batch, channels* imagePixels.
   void shrinkMat(const MatrixPtr& in, MatrixPtr& out);
 
+  void onPassEnd() { firstTest_ = true; }
+
   MatrixPtr tmpMat_, tmpGrad_;
   MatrixPtr expandedIn_, expandedOut_;
   MatrixPtr expandedInGrad_, expandedOutGrad_, inGrad_;