- added setBatch()

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@771 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-10 10:36:55 +00:00
parent e240eae073
commit 648d8a002e
6 changed files with 19 additions and 8 deletions
@@ -204,7 +204,8 @@ void Stack::train(Rbm::IListener* pListener)
 	while(pLayer)
 	{
 		std::cout << m_name << ": " << " Training of layer " << std::to_string(pLayer->id()) << std::endl;
-		pLayer->train(m_trainingData, pListener);
+		pLayer->setBatch(m_trainingData);
+		pLayer->train(pListener);
 		pLayer = pLayer->next;
 	}
 }