- no weighInit in prj
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@600 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+5
-2
@@ -105,16 +105,19 @@ int main()
|
||||
stack.save();
|
||||
|
||||
// Shake weights
|
||||
stack.weightsInit(0.01);
|
||||
stack.saveWeights();
|
||||
|
||||
#else
|
||||
// Load project
|
||||
stack.load();
|
||||
|
||||
// Load weights
|
||||
stack.loadWeights();
|
||||
#endif
|
||||
|
||||
// Train stack
|
||||
stack.train(batch, 1000, 1000, &statusDisplay);
|
||||
stack.weightsInit(0.01);
|
||||
stack.train(batch, 100, 1000, &statusDisplay);
|
||||
|
||||
// Save weights
|
||||
stack.saveWeights();
|
||||
|
||||
Reference in New Issue
Block a user