- cleaned up
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@828 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
+11
-2
@@ -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 ¶m)
|
||||
{
|
||||
Layer *pLayer = getLayer(0);
|
||||
while (pLayer)
|
||||
{
|
||||
pLayer->params() = param;
|
||||
pLayer = pLayer->next;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private:
|
||||
size_t m_numContext;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user