- load and save of weights

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@588 b431acfa-c32f-4a4a-93f1-934dc6c82436

source/Stack.cpp

@@ -65,7 +65,7 @@ RbmLayer* Stack::getLayer(size_t layerId) const
 	
 }
 
-void Stack::save(size_t numTraining)
+bool Stack::save(size_t numTraining)
 {
 	std::cout << "Exporting Project " << m_prjname << std::endl;
 	ofstream ofs(m_prjname + string(".prj"));
@@ -86,16 +86,36 @@ void Stack::save(size_t numTraining)
 	project["stack"]["layers"] = layers;
 	
 	ofs << writer.write(project);
+	
+	return true;
 }
 
-void Stack::saveWeights()
+bool Stack::loadWeights()
 {
 	RbmLayer *pLayer = m_pLayers;
 	while(pLayer)
 	{
-		pLayer->saveWeights();
+		if (!pLayer->loadWeights())
+		{
+			return false;
+		}
 		pLayer = pLayer->upper;
 	}
+	return true;
+}
+
+bool Stack::saveWeights()
+{
+	RbmLayer *pLayer = m_pLayers;
+	while(pLayer)
+	{
+		if (!pLayer->saveWeights())
+		{
+			return false;
+		}
+		pLayer = pLayer->upper;
+	}
+	return true;
 }
 
 void Stack::train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs, Rbm::IListener* pListener)