- improved poet

- fixed time reversal in Rnn train and forward

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@834 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-18 17:56:43 +00:00
parent a061a94099
commit 0544e7721b
6 changed files with 224 additions and 245 deletions
-22
View File
@@ -159,31 +159,12 @@ int main()
stack->loadTrainingBatch(".");
arma::mat t_vc = stack->trainingBatch();
Rbm::Params params = stack->getLayer(0)->params();
#if DO_TRAINING
int numGibbs = params.numGibbs;
params.numGibbs = 1;
stack->setParams(params);
RbmListener listener;
// Phase 10
params.numEpochs = 2000;
stack->setParams(params);
stack->train(t_vc, &listener);
#if 0
// Phase 11
params.numEpochs = 10000;
stack->setParams(params);
stack->train(t_vc, &listener);
// Phase 20
params.numGibbs = numGibbs;
stack->setParams(params);
stack->train(t_vc, &listener);
#endif
stack->saveWeights(".");
stack->save(".");
@@ -207,9 +188,6 @@ int main()
#endif
#if DO_FORWARD
params.gibbsDoSampleHidden = false;
params.gibbsDoSampleVisible = false;
stack->setParams(params);
arma::mat state;
arma::mat curr;
for (int i=0; i < t_vc.n_rows; i++)