Skip to content
Snippets Groups Projects
Commit 6bd51e45 authored by Karel Vesely's avatar Karel Vesely
Browse files

trunk,nnet1: bugfix, adding missing setter method


git-svn-id: https://svn.code.sf.net/p/kaldi/code/trunk@4473 5e6a8d80-dfce-4ca6-a32a-6e07a63d50c8
parent 574479ca
Branches
No related tags found
No related merge requests found
...@@ -492,6 +492,9 @@ class AddShift : public UpdatableComponent { ...@@ -492,6 +492,9 @@ class AddShift : public UpdatableComponent {
KALDI_ASSERT(shift_data.Dim() == shift_data_.Dim()); KALDI_ASSERT(shift_data.Dim() == shift_data_.Dim());
shift_data_.CopyFromVec(shift_data); shift_data_.CopyFromVec(shift_data);
} }
void SetLearnRateCoef(float c) {
learn_rate_coef_ = c;
}
protected: protected:
CuVector<BaseFloat> shift_data_; CuVector<BaseFloat> shift_data_;
...@@ -599,6 +602,9 @@ class Rescale : public UpdatableComponent { ...@@ -599,6 +602,9 @@ class Rescale : public UpdatableComponent {
KALDI_ASSERT(scale_data.Dim() == scale_data_.Dim()); KALDI_ASSERT(scale_data.Dim() == scale_data_.Dim());
scale_data_.CopyFromVec(scale_data); scale_data_.CopyFromVec(scale_data);
} }
void SetLearnRateCoef(float c) {
learn_rate_coef_ = c;
}
protected: protected:
CuVector<BaseFloat> scale_data_; CuVector<BaseFloat> scale_data_;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment