- init weights on model load error

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@830 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-18 15:36:04 +00:00
parent 36010b12eb
commit 87fc780461
2 changed files with 10 additions and 10 deletions
+4 -7
View File
@@ -138,8 +138,8 @@ arma::mat createTraining(const string &filename, size_t seq_len)
}
#define CREATE_TRAINING 0
#define DO_TRAINING 0
#define DO_FORWARD 1
#define DO_TRAINING 1
#define DO_FORWARD 0
int main()
{
@@ -167,16 +167,12 @@ int main()
stack->setParams(params);
RbmListener listener;
for (int i=0; i < stack->getSeqLen(); i++)
{
stack->getLayer(i)->weightsInit(0.01,0);
}
// Phase 10
params.numEpochs = 2000;
stack->setParams(params);
stack->train(t_vc, &listener);
#if 0
// Phase 11
params.numEpochs = 10000;
stack->setParams(params);
@@ -186,6 +182,7 @@ int main()
params.numGibbs = numGibbs;
stack->setParams(params);
stack->train(t_vc, &listener);
#endif
stack->saveWeights(".");
stack->save(".");