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

- rbm: no training params at construction

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@591 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 19:20:34 +00:00
parent 711ad3057e
commit e2bd0ee850
6 changed files with 13 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/main.cpp
+3 -4
View File
@@ -82,7 +82,6 @@ int main()
const string project("mnist_2");
RbmListener statusDisplay;
Rbm::Params rbmParams;
Stack stack(project);
arma::mat batch = loadTraining(project + string(".training.dat"));
@@ -94,16 +93,16 @@ int main()
printf("Loaded %d training samples\n", (int)numTraining);
#if 0
#if 1
int i=0;
RbmLayer *lowerLayer = new RbmLayer("Layer", i, numVisibleX, numVisibleY, numHidden, rbmParams);
RbmLayer *lowerLayer = new RbmLayer("Layer", i, numVisibleX, numVisibleY, numHidden);
stack.addLayer(lowerLayer);
numHidden >>= 1;
i++;
for (i; i < 4; i++)
{
RbmLayer *layer = new RbmLayer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden, rbmParams);
RbmLayer *layer = new RbmLayer("Layer", i, lowerLayer->bh().n_elem, 1, numHidden);
lowerLayer = layer;
stack.addLayer(layer);
numHidden >>= 1;