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
This commit is contained in:
2019-11-07 07:39:13 +00:00
parent 1025fc82c1
commit 10aff804f0
2 changed files with 5 additions and 1 deletions
+3
View File
@@ -11,6 +11,7 @@
* Created on 21. Oktober 2019, 21:28
*/
#include <cassert>
#include "Rbm.hpp"
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_bh(1, numHidden)
{
assert(numVisible > 0);
assert(numHidden > 0);
Noise_Init(&m_noise, 0x32727155);
}
+2 -1
View File
@@ -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;