From 7a9453594b9395a56d21259e5793206c00dd1a05 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Mon, 28 Oct 2019 20:15:45 +0000 Subject: [PATCH] - more compact creation of Stack git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@595 b431acfa-c32f-4a4a-93f1-934dc6c82436 --- source/main.cpp | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/source/main.cpp b/source/main.cpp index a2e9be9..8779d85 100644 --- a/source/main.cpp +++ b/source/main.cpp @@ -90,22 +90,15 @@ int main() printf("Loaded %d training samples\n", (int)numTraining); #if 0 - size_t numVisibleX = 16; - size_t numVisibleY = 16; - size_t numHidden = 256; - - int i=0; - Layer *lowerLayer = new Layer("Layer", i, numVisibleX, numVisibleY, numHidden); + int i = 0; + Layer *lowerLayer = new Layer("Layer", i, 16, 16, 256); stack.addLayer(lowerLayer); - numHidden >>= 1; - i++; - for (i; i < 1; i++) + for (++i; i < 1; i++) { - Layer *layer = new Layer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden); + Layer *layer = new Layer("Layer", i, lowerLayer->bh().n_elem, 1, lowerLayer->bh().n_elem >> 1); lowerLayer = layer; stack.addLayer(layer); - numHidden >>= 1; } // Save project