From c10d9c6e38e1e72e634bb383bfb050f775e7c0b1 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Fri, 21 Jan 2022 06:28:39 +0000 Subject: [PATCH] - 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 --- source/AStack.cpp | 9 ++++----- source/poet.cpp | 5 +---- 2 files changed, 5 insertions(+), 9 deletions(-) diff --git a/source/AStack.cpp b/source/AStack.cpp index d1e6b31..442d749 100644 --- a/source/AStack.cpp +++ b/source/AStack.cpp @@ -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() diff --git a/source/poet.cpp b/source/poet.cpp index 431e55e..93b5b6e 100644 --- a/source/poet.cpp +++ b/source/poet.cpp @@ -133,10 +133,7 @@ int main(int argc, char *argv[]) if (command == Command::Reset) { - for (int i=0; i < stack->numLayers(); i++) - { - stack->getLayer(i)->weightsInit(0.01); - } + stack->weightsInit(0.01); stack->saveWeights("."); }