- 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:
@@ -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++)
|
||||
|
||||
Reference in New Issue
Block a user