- weightsInit only on create
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@603 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+3
-1
@@ -105,6 +105,9 @@ int main()
|
||||
stack.save();
|
||||
|
||||
// Shake weights
|
||||
stack.weightsInit(0.01);
|
||||
|
||||
// Save weights
|
||||
stack.saveWeights();
|
||||
|
||||
#else
|
||||
@@ -116,7 +119,6 @@ int main()
|
||||
#endif
|
||||
|
||||
// Train stack
|
||||
stack.weightsInit(0.01);
|
||||
stack.train(batch, 100, 1000, &statusDisplay);
|
||||
|
||||
// Save weights
|
||||
|
||||
Reference in New Issue
Block a user