- poet use AStack::weightsInit

- AStack::weightsSave continue for other layer on failure git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@854 b431acfa-c32f-4a4a-93f1-934dc6c82436
2022-01-21 06:28:39 +00:00
parent 02b40561ec
commit c10d9c6e38
2 changed files with 5 additions and 9 deletions
@@ -132,16 +132,15 @@ bool AStack::loadWeights(const std::string &dir)

 bool AStack::saveWeights(const std::string &dir)
 {
+	bool result = true;
 	Layer *pLayer = m_pLayers;
 	while(pLayer)
 	{
-		if (!pLayer->weightsSave(dir, m_name))
-		{
-			return false;
-		}
+		bool success = pLayer->weightsSave(dir, m_name);
+		result &= success;
 		pLayer = pLayer->next;
 	}
-	return true;
+	return result;
 }

 arma::mat& AStack::trainingBatch()