Rbm: added assertion check for params

- improved creation of stack git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@609 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-11-07 07:39:13 +00:00
parent 1025fc82c1
commit 10aff804f0
2 changed files with 5 additions and 1 deletions
@@ -90,11 +90,12 @@ int main()
 	printf("Loaded %d training samples\n", (int)numTraining);

 #if 0
+	const int numLayers = 4;
 	int i = 0;
 	Layer *lowerLayer = new Layer("Layer", i, 16, 16, 8);
 	stack.addLayer(lowerLayer);
 	
-	for (++i; i < 4; i++)
+	for (++i; i < numLayers; i++)
 	{
 		Layer *layer = new Layer("Layer", i, lowerLayer->bh().n_elem, 1, lowerLayer->bh().n_elem >> 1);
 		layer->params().learningRate = lowerLayer->params().learningRate/2;