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

use explicit relative imports for aidatatang_200zh

Browse Source
This commit is contained in:
shaynemei 2022-08-01 20:54:59 -07:00
parent 60621b242a
commit 7daf148342
5 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/aidatatang_200zh/ASR/local/test_prepare_lang.py
View File

@ -22,7 +22,7 @@ import os
import tempfile
import k2
from prepare_lang import (
from .prepare_lang import (
add_disambig_symbols,
generate_id_map,
get_phones,

6
egs/aidatatang_200zh/ASR/pruned_transducer_stateless2/decode.py
View File

@ -59,15 +59,15 @@ from typing import Dict, List, Optional, Tuple
import k2
import torch
import torch.nn as nn
from asr_datamodule import Aidatatang_200zhAsrDataModule
from beam_search import (
from .asr_datamodule import Aidatatang_200zhAsrDataModule
from .beam_search import (
beam_search,
fast_beam_search_one_best,
greedy_search,
greedy_search_batch,
modified_beam_search,
)
from train import get_params, get_transducer_model
from .train import get_params, get_transducer_model
from icefall.checkpoint import (
average_checkpoints,

2
egs/aidatatang_200zh/ASR/pruned_transducer_stateless2/export.py
View File

@ -46,7 +46,7 @@ import logging
from pathlib import Path
import torch
from train import get_params, get_transducer_model
from .train import get_params, get_transducer_model
from icefall.checkpoint import average_checkpoints, load_checkpoint
from icefall.lexicon import Lexicon

4
egs/aidatatang_200zh/ASR/pruned_transducer_stateless2/pretrained.py
View File

@ -63,7 +63,7 @@ import k2
import kaldifeat
import torch
import torchaudio
from beam_search import (
from .beam_search import (
beam_search,
fast_beam_search_one_best,
greedy_search,
@ -71,7 +71,7 @@ from beam_search import (
modified_beam_search,
)
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

12
egs/aidatatang_200zh/ASR/pruned_transducer_stateless2/train.py
View File

@ -57,15 +57,15 @@ import optim
import torch
import torch.multiprocessing as mp
import torch.nn as nn
from asr_datamodule import Aidatatang_200zhAsrDataModule
from conformer import Conformer
from decoder import Decoder
from joiner import Joiner
from .asr_datamodule import Aidatatang_200zhAsrDataModule
from .conformer import Conformer
from .decoder import Decoder
from .joiner import Joiner
from lhotse.cut import Cut
from lhotse.dataset.sampling.base import CutSampler
from lhotse.utils import fix_random_seed
from model import Transducer
from optim import Eden, Eve
from .model import Transducer
from .optim import Eden, Eve
from torch import Tensor
from torch.cuda.amp import GradScaler
from torch.nn.parallel import DistributedDataParallel as DDP