From 9a47b6219a76976b53863b6fab09a9d53eb710e4 Mon Sep 17 00:00:00 2001
From: dangqingqing <dangqingqing@baidu.com>
Date: Thu, 22 Dec 2016 20:32:28 +0800
Subject: [PATCH] Bug fix for BatchNorm when testing more than models in test
 mode.

---
 paddle/gserver/layers/BatchNormalizationLayer.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/gserver/layers/BatchNormalizationLayer.h b/paddle/gserver/layers/BatchNormalizationLayer.h
index 052c207732..5e3e697bce 100644
--- a/paddle/gserver/layers/BatchNormalizationLayer.h
+++ b/paddle/gserver/layers/BatchNormalizationLayer.h
@@ -58,6 +58,8 @@ protected:
   /// 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_;
-- 
GitLab