mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 22:54:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-03-23 17:35:39 +08:00
parent 127155b37d
commit 055406a67e
4 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/aishell/ASR/transducer_stateless_modified-2/pretrained.py
View File

@ -55,7 +55,6 @@ from typing import List
import kaldifeat import kaldifeat
import torch import torch
import torch.nn as nn
import torchaudio import torchaudio
from beam_search import ( from beam_search import (
beam_search, beam_search,
@ -67,7 +66,6 @@ from torch.nn.utils.rnn import pad_sequence
from train import get_params, get_transducer_model from train import get_params, get_transducer_model
from icefall.lexicon import Lexicon from icefall.lexicon import Lexicon
from icefall.utils import AttributeDict
def get_parser(): def get_parser():
@ -262,7 +260,9 @@ def main():
) )
hyp_list.append(hyp) hyp_list.append(hyp)
hyps = [sp.decode(hyp).split() for hyp in hyp_list] hyps = []
for hyp in hyp_list:
hyps.append([lexicon.token_table[i] for i in hyp])
s = "\n" s = "\n"
for filename, hyp in zip(params.sound_files, hyps): for filename, hyp in zip(params.sound_files, hyps):

6
egs/aishell/ASR/transducer_stateless_modified/pretrained.py
View File

@ -55,7 +55,6 @@ from typing import List
import kaldifeat import kaldifeat
import torch import torch
import torch.nn as nn
import torchaudio import torchaudio
from beam_search import ( from beam_search import (
beam_search, beam_search,
@ -67,7 +66,6 @@ from torch.nn.utils.rnn import pad_sequence
from train import get_params, get_transducer_model from train import get_params, get_transducer_model
from icefall.lexicon import Lexicon from icefall.lexicon import Lexicon
from icefall.utils import AttributeDict
def get_parser(): def get_parser():
@ -262,7 +260,9 @@ def main():
) )
hyp_list.append(hyp) hyp_list.append(hyp)
hyps = [sp.decode(hyp).split() for hyp in hyp_list] hyps = []
for hyp in hyp_list:
hyps.append([lexicon.token_table[i] for i in hyp])
s = "\n" s = "\n"
for filename, hyp in zip(params.sound_files, hyps): for filename, hyp in zip(params.sound_files, hyps):

3
egs/librispeech/ASR/transducer_stateless/pretrained.py
View File

@ -59,7 +59,6 @@ from typing import List
import kaldifeat import kaldifeat
import sentencepiece as spm import sentencepiece as spm
import torch import torch
import torch.nn as nn
import torchaudio import torchaudio
from beam_search import ( from beam_search import (
beam_search, beam_search,
@ -70,8 +69,6 @@ from beam_search import (
from torch.nn.utils.rnn import pad_sequence from torch.nn.utils.rnn import pad_sequence
from train import get_params, get_transducer_model from train import get_params, get_transducer_model
from icefall.utils import AttributeDict
def get_parser(): def get_parser():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(

3
egs/librispeech/ASR/transducer_stateless_multi_datasets/pretrained.py
View File

@ -59,7 +59,6 @@ from typing import List
import kaldifeat import kaldifeat
import sentencepiece as spm import sentencepiece as spm
import torch import torch
import torch.nn as nn
import torchaudio import torchaudio
from beam_search import ( from beam_search import (
beam_search, beam_search,
@ -70,8 +69,6 @@ from beam_search import (
from torch.nn.utils.rnn import pad_sequence from torch.nn.utils.rnn import pad_sequence
from train import get_params, get_transducer_model from train import get_params, get_transducer_model
from icefall.utils import AttributeDict
def get_parser(): def get_parser():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(