- more compact

- added Stack::weights init
- create smaller test.prj

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@596 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2019-10-28 22:53:29 +00:00
parent 7a9453594b
commit 595be173b1
5 changed files with 22 additions and 9 deletions
+8 -5
View File
@@ -89,9 +89,9 @@ int main()
size_t numTraining = batch.n_rows;
printf("Loaded %d training samples\n", (int)numTraining);
#if 0
#if 1
int i = 0;
Layer *lowerLayer = new Layer("Layer", i, 16, 16, 256);
Layer *lowerLayer = new Layer("Layer", i, 16, 16, 8);
stack.addLayer(lowerLayer);
for (++i; i < 1; i++)
@@ -103,13 +103,16 @@ int main()
// Save project
stack.save();
// Shake weights
stack.weightsInit(0.01);
#else
// Load project
stack.load();
#endif
// Load weights
stack.loadWeights();
#endif
// Train stack
stack.train(batch, 1000, 1000, &statusDisplay);