[src] Bug fix: LDA output dim should be LDA matrix dim, not input feature dim (#4003)

nshmyrev · web-flow · commit b1d809386c17 · 2020-03-20T16:19:41.000+08:00
diff --git a/src/cudafeat/online-ivector-feature-cuda.cc b/src/cudafeat/online-ivector-feature-cuda.cc
@@ -45,7 +45,7 @@ void IvectorExtractorFastCuda::GetIvector(const CuMatrixBase<BaseFloat> &feats,
   int lda_cols = cu_lda_.NumCols();
 
   // normalized pipeline
-  CuMatrix<BaseFloat> lda_feats_normalized(rows, cols, kUndefined);
+  CuMatrix<BaseFloat> lda_feats_normalized(rows, lda_rows, kUndefined);
   {
     CudaOnlineCmvn cmvn(info_.cmvn_opts, naive_cmvn_state_);
     CuMatrix<BaseFloat> cmvn_feats(rows, cols, kUndefined);
@@ -81,7 +81,7 @@ void IvectorExtractorFastCuda::GetIvector(const CuMatrixBase<BaseFloat> &feats,
   }
 
   // non-normalized pipeline
-  CuMatrix<BaseFloat> lda_feats(feats.NumRows(), feats.NumCols(), kUndefined);
+  CuMatrix<BaseFloat> lda_feats(rows, lda_rows, kUndefined);
   {
     CuMatrix<BaseFloat> spliced_feats;
 

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ void IvectorExtractorFastCuda::GetIvector(const CuMatrixBase<BaseFloat> &feats,`
`45`	`45`	`int lda_cols = cu_lda_.NumCols();`
`46`	`46`
`47`	`47`	`// normalized pipeline`
`48`		`- CuMatrix<BaseFloat> lda_feats_normalized(rows, cols, kUndefined);`
	`48`	`+ CuMatrix<BaseFloat> lda_feats_normalized(rows, lda_rows, kUndefined);`
`49`	`49`	`{`
`50`	`50`	`CudaOnlineCmvn cmvn(info_.cmvn_opts, naive_cmvn_state_);`
`51`	`51`	`CuMatrix<BaseFloat> cmvn_feats(rows, cols, kUndefined);`
`@@ -81,7 +81,7 @@ void IvectorExtractorFastCuda::GetIvector(const CuMatrixBase<BaseFloat> &feats,`
`81`	`81`	`}`
`82`	`82`
`83`	`83`	`// non-normalized pipeline`
`84`		`- CuMatrix<BaseFloat> lda_feats(feats.NumRows(), feats.NumCols(), kUndefined);`
	`84`	`+ CuMatrix<BaseFloat> lda_feats(rows, lda_rows, kUndefined);`
`85`	`85`	`{`
`86`	`86`	`CuMatrix<BaseFloat> spliced_feats;`
`87`	`87`