- more compact
- added Stack::weights init - create smaller test.prj git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@596 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+1
-2
@@ -110,8 +110,7 @@ void Rbm::train(const arma::mat& batch, size_t miniBatchSize, size_t numEpochs,
|
||||
vis_state = miniBatch;
|
||||
}
|
||||
|
||||
hid_state = toHiddenState(vis_state);
|
||||
hid_probs = probsLogistic(hid_state);
|
||||
hid_probs = probsLogistic(toHiddenState(vis_state));
|
||||
|
||||
// Sample hidden
|
||||
if (m_params.doRaoBlackwell)
|
||||
|
||||
Reference in New Issue
Block a user