- 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:
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user