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

avx_docs
dangqingqing 8 years ago
parent 4f70880372
commit 9a47b6219a

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

Loading…
Cancel
Save