diff --git a/source/Rbm.cpp b/source/Rbm.cpp index b4d7332..ac33801 100644 --- a/source/Rbm.cpp +++ b/source/Rbm.cpp @@ -176,7 +176,6 @@ void Rbm::train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, arma::mat diffErr = miniBatch - vis_probs; arma::mat diffErr_squared = diffErr % diffErr; status.err = accu(diffErr_squared)/diffErr_squared.n_elem; - status.err_total = 0; status.progress += dProgress*miniBatchSizeActual; if (pListener) diff --git a/source/Rbm.hpp b/source/Rbm.hpp index 0671f43..10f906a 100644 --- a/source/Rbm.hpp +++ b/source/Rbm.hpp @@ -86,7 +86,7 @@ public: , trainingSizeRemain(0) , progress(0) , err(-1.0) - , err_total(1-0) + , err_total(-1.0) , L1(-1.0) , L2(-1.0) {