| | |
|---|
| | | ) -> List: |
|---|
| | | # make hotword list |
|---|
| | | hotwords, hotwords_length = self.proc_hotword(hotwords) |
|---|
| | | bias_embed = self.eb_infer(torch.Tensor(hotwords)) |
|---|
| | | if int(self.device_id) != -1: |
|---|
| | | bias_embed = self.eb_infer(hotwords.cuda()) |
|---|
| | | else: |
|---|
| | | bias_embed = self.eb_infer(hotwords) |
|---|
| | | # index from bias_embed |
|---|
| | | bias_embed = torch.transpose(bias_embed, 0, 1) |
|---|
| | | _ind = np.arange(0, len(hotwords)).tolist() |
|---|
| | |
|---|
| | | outputs = self.bb_infer(feats, feats_len, bias_embed) |
|---|
| | | am_scores, valid_token_lens = outputs[0], outputs[1] |
|---|
| | | else: |
|---|
| | | outputs = self.bb_infer_infer(feats.cuda(), feats_len.cuda(), bias_embed) |
|---|
| | | outputs = self.bb_infer_infer(feats.cuda(), feats_len.cuda(), bias_embed.cuda()) |
|---|
| | | am_scores, valid_token_lens = outputs[0].cpu(), outputs[1].cpu() |
|---|
| | | except: |
|---|
| | | # logging.warning(traceback.format_exc()) |
|---|
| | |
|---|
| | | hotword_int = [word_map(i) for i in hotwords] |
|---|
| | | hotword_int.append(np.array([1])) |
|---|
| | | hotwords = pad_list(hotword_int, pad_value=0, max_len=10) |
|---|
| | | return hotwords, hotwords_length |
|---|
| | | return torch.tensor(hotwords), hotwords_length |
|---|
| | | |
|---|
| | | def bb_infer( |
|---|
| | | self, feats, feats_len, bias_embed |
|---|
