- 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();
|
stack.save();
|
||||||
|
|
||||||
// Shake weights
|
// Shake weights
|
||||||
|
stack.weightsInit(0.01);
|
||||||
|
|
||||||
|
// Save weights
|
||||||
stack.saveWeights();
|
stack.saveWeights();
|
||||||
|
|
||||||
#else
|
#else
|
||||||
@@ -116,7 +119,6 @@ int main()
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Train stack
|
// Train stack
|
||||||
stack.weightsInit(0.01);
|
|
||||||
stack.train(batch, 100, 1000, &statusDisplay);
|
stack.train(batch, 100, 1000, &statusDisplay);
|
||||||
|
|
||||||
// Save weights
|
// Save weights
|
||||||
|
|||||||
Reference in New Issue
Block a user