jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- moved numEpochs and miniBatchSize to RBM::Params

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@606 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-11-06 06:12:51 +00:00
parent c834a86f0f
commit 3ef7b07f5e
6 changed files with 31 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Stack.cpp
+4 -4
View File
@@ -153,17 +153,17 @@ bool Stack::saveWeights()
return true;
}
void Stack::train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener)
void Stack::train(const arma::mat& batch, Rbm::IListener* pListener)
{
Layer *pLayer = m_pLayers;
while(pLayer)
{
train(pLayer->id(), batch, miniBatchSize, numEpochs, pListener);
train(pLayer->id(), batch, pListener);
pLayer = pLayer->upper;
}
}
void Stack::train(size_t layerId, const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener)
void Stack::train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener)
{
arma::mat thisBatch = batch;
Layer *pLayer = m_pLayers;
@@ -180,7 +180,7 @@ void Stack::train(size_t layerId, const arma::mat& batch, size_t miniBatchSize,
if (pLayer)
{
std::cout << m_prjname << ": " << " Training of layer " << std::to_string(layerId) << std::endl;
pLayer->train(thisBatch, miniBatchSize, numEpochs, pListener);
pLayer->train(thisBatch, pListener);
}
}