提交 16d16b49 编写于 作者: L LCY-Seso 提交者: GitHub

Merge pull request #999 from qingqing01/bug_fix

Bug fix for BatchNorm when testing more than models in test mode.
...@@ -58,6 +58,8 @@ protected: ...@@ -58,6 +58,8 @@ protected:
/// to batch, channels* imagePixels. /// to batch, channels* imagePixels.
void shrinkMat(const MatrixPtr& in, MatrixPtr& out); void shrinkMat(const MatrixPtr& in, MatrixPtr& out);
void onPassEnd() { firstTest_ = true; }
MatrixPtr tmpMat_, tmpGrad_; MatrixPtr tmpMat_, tmpGrad_;
MatrixPtr expandedIn_, expandedOut_; MatrixPtr expandedIn_, expandedOut_;
MatrixPtr expandedInGrad_, expandedOutGrad_, inGrad_; MatrixPtr expandedInGrad_, expandedOutGrad_, inGrad_;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册