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

Rbm: added assertion check for params

Browse Source 
- improved creation of stack

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@609 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-11-07 07:39:13 +00:00
parent 1025fc82c1
commit 10aff804f0
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Rbm.cpp
+3
View File
@@ -11,6 +11,7 @@
* Created on 21. Oktober 2019, 21:28 * Created on 21. Oktober 2019, 21:28
*/ */
#include <cassert>
#include "Rbm.hpp" #include "Rbm.hpp"
Rbm::Rbm(size_t numVisible, size_t numHidden) Rbm::Rbm(size_t numVisible, size_t numHidden)
@@ -19,6 +20,8 @@ Rbm::Rbm(size_t numVisible, size_t numHidden)
, m_bv(1, numVisible) , m_bv(1, numVisible)
, m_bh(1, numHidden) , m_bh(1, numHidden)
{ {
assert(numVisible > 0);
assert(numHidden > 0);
Noise_Init(&m_noise, 0x32727155); Noise_Init(&m_noise, 0x32727155);
} }
source/main.cpp
+2 -1
View File
@@ -90,11 +90,12 @@ int main()
printf("Loaded %d training samples\n", (int)numTraining); printf("Loaded %d training samples\n", (int)numTraining);
#if 0 #if 0
const int numLayers = 4;
int i = 0; int i = 0;
Layer *lowerLayer = new Layer("Layer", i, 16, 16, 8); Layer *lowerLayer = new Layer("Layer", i, 16, 16, 8);
stack.addLayer(lowerLayer); 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 *layer = new Layer("Layer", i, lowerLayer->bh().n_elem, 1, lowerLayer->bh().n_elem >> 1);
layer->params().learningRate = lowerLayer->params().learningRate/2; layer->params().learningRate = lowerLayer->params().learningRate/2;