- improved progressIndicator
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@639 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+1
-1
@@ -96,7 +96,6 @@ void Rbm::train(const arma::mat& batch, IListener* pListener)
|
|||||||
for (int epoch=0; epoch < m_params.numEpochs; epoch++)
|
for (int epoch=0; epoch < m_params.numEpochs; epoch++)
|
||||||
{
|
{
|
||||||
|
|
||||||
status.progress = (int)(progress + 0.5);
|
|
||||||
if (status.progress != lastProgress)
|
if (status.progress != lastProgress)
|
||||||
{
|
{
|
||||||
lastProgress = status.progress;
|
lastProgress = status.progress;
|
||||||
@@ -181,6 +180,7 @@ void Rbm::train(const arma::mat& batch, IListener* pListener)
|
|||||||
m_w += learning_rate*momentum_weights;
|
m_w += learning_rate*momentum_weights;
|
||||||
|
|
||||||
progress += dProgress*miniBatchSizeActual;
|
progress += dProgress*miniBatchSizeActual;
|
||||||
|
status.progress = (int)(progress + 0.5);
|
||||||
|
|
||||||
} // Number of epochs
|
} // Number of epochs
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user