jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- read/write of Stack project

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@589 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 18:55:12 +00:00
parent 9bfee1ae16
commit 43b475eac1
5 changed files with 71 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/main.cpp
+7 -3
View File
@@ -94,22 +94,26 @@ int main()
printf("Loaded %d training samples\n", (int)numTraining);
#if 0
int i=0;
RbmLayer *lowerLayer = new RbmLayer(project, i, numVisibleX, numVisibleY, numHidden, rbmParams);
RbmLayer *lowerLayer = new RbmLayer("Layer", i, numVisibleX, numVisibleY, numHidden, rbmParams);
stack.addLayer(lowerLayer);
numHidden >>= 1;
i++;
for (i; i < 4; i++)
{
RbmLayer *layer = new RbmLayer(project, i, lowerLayer->bh().n_elem, 1, numHidden, rbmParams);
RbmLayer *layer = new RbmLayer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden, rbmParams);
lowerLayer = layer;
stack.addLayer(layer);
numHidden >>= 1;
}
// Save project
stack.save(numTraining);
stack.save();
#else
stack.load();
#endif
// Load weights
stack.loadWeights();