- cleaned up

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@828 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-18 15:22:28 +00:00
parent 9a8ad70a0b
commit d6ea343d4a
6 changed files with 44 additions and 172 deletions
+11 -2
View File
@@ -25,9 +25,7 @@ public:
RnnStack(const RnnStack& orig) = delete;
virtual ~RnnStack();
arma::mat trainingBatchFrom(size_t layerId, const arma::mat& batch) override;
void train(const arma::mat& batch, Rbm::IListener* pListener) override;
void train(size_t layerId, const arma::mat& batch, Rbm::IListener* pListener) override;
size_t getSeqLen();
size_t numContext();
@@ -48,6 +46,17 @@ public:
return v.cols(NUM_CODES, 2*NUM_CODES-1);
}
void setParams(Rbm::Params const &param)
{
Layer *pLayer = getLayer(0);
while (pLayer)
{
pLayer->params() = param;
pLayer = pLayer->next;
}
}
private:
size_t m_numContext;
};