diff --git a/source/Rbm.cpp b/source/Rbm.cpp index a5e4b4f..0099164 100644 --- a/source/Rbm.cpp +++ b/source/Rbm.cpp @@ -96,7 +96,6 @@ void Rbm::train(const arma::mat& batch, IListener* pListener) for (int epoch=0; epoch < m_params.numEpochs; epoch++) { - status.progress = (int)(progress + 0.5); if (status.progress != lastProgress) { lastProgress = status.progress; @@ -181,6 +180,7 @@ void Rbm::train(const arma::mat& batch, IListener* pListener) m_w += learning_rate*momentum_weights; progress += dProgress*miniBatchSizeActual; + status.progress = (int)(progress + 0.5); } // Number of epochs