- load and save of weights

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@588 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2019-10-28 17:26:59 +00:00
parent 27c07088c2
commit 9bfee1ae16
6 changed files with 114 additions and 22 deletions
+9
View File
@@ -107,8 +107,17 @@ int main()
stack.addLayer(layer);
numHidden >>= 1;
}
// Save project
stack.save(numTraining);
// Load weights
stack.loadWeights();
// Train stack
stack.train(batch, 1000, 100, &statusDisplay);
// Save weights
stack.saveWeights();
RbmLayer *layer = stack.getLayer(0);