Merge pull request #999 from qingqing01/bug_fix

Bug fix for BatchNorm when testing more than models in test mode.
avx_docs
LCY-Seso 8 years ago committed by GitHub
commit 16d16b4921

@ -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_;

Loading…
Cancel
Save