funasr/models/e2e_vad.py
@@ -311,7 +311,7 @@ 0.000001)) def ComputeScores(self, feats: torch.Tensor, in_cache: Dict[str, torch.Tensor]) -> None: scores = self.encoder(feats, in_cache) # return B * T * D scores = self.encoder(feats, in_cache).to('cpu') # return B * T * D assert scores.shape[1] == feats.shape[1], "The shape between feats and scores does not match" self.vad_opts.nn_eval_block_size = scores.shape[1] self.frm_cnt += scores.shape[1] # count total frames
