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 aishell4

Browse Source
This commit is contained in:
shaynemei 2022-08-01 21:37:24 -07:00
parent 4cb62a3202
commit 6df69603a8
7 changed files with 18 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/aishell4/ASR/local/prepare_char.py
View File

@ -39,7 +39,7 @@ from typing import Dict, List
import k2
import torch
from prepare_lang import (
from .prepare_lang import (
Lexicon,
add_disambig_symbols,
add_self_loops,

2
egs/aishell4/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,

8
egs/aishell4/ASR/pruned_transducer_stateless5/decode.py
View File

@ -61,8 +61,8 @@ from typing import Dict, List, Optional, Tuple
import k2
import torch
import torch.nn as nn
from asr_datamodule import Aishell4AsrDataModule
from beam_search import (
from .asr_datamodule import Aishell4AsrDataModule
from .beam_search import (
beam_search,
fast_beam_search_one_best,
greedy_search,
@ -70,8 +70,8 @@ from beam_search import (
modified_beam_search,
)
from lhotse.cut import Cut
from local.text_normalize import text_normalize
from train import add_model_arguments, get_params, get_transducer_model
from ..local.text_normalize import text_normalize
from .train import add_model_arguments, get_params, get_transducer_model
from icefall.checkpoint import (
average_checkpoints,

2
egs/aishell4/ASR/pruned_transducer_stateless5/export.py
View File

@ -49,7 +49,7 @@ import logging
from pathlib import Path
import torch
from train import add_model_arguments, get_params, get_transducer_model
from .train import add_model_arguments, get_params, get_transducer_model
from icefall.checkpoint import (
average_checkpoints,

4
egs/aishell4/ASR/pruned_transducer_stateless5/pretrained.py
View File

@ -72,7 +72,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,
@ -80,7 +80,7 @@ from beam_search import (
modified_beam_search,
)
from torch.nn.utils.rnn import pad_sequence
from train import add_model_arguments, get_params, get_transducer_model
from .train import add_model_arguments, get_params, get_transducer_model
from icefall.lexicon import Lexicon

2
egs/aishell4/ASR/pruned_transducer_stateless5/test_model.py
View File

@ -23,7 +23,7 @@ To run this file, do:
python ./pruned_transducer_stateless5/test_model.py
"""
from train import get_params, get_transducer_model
from .train import get_params, get_transducer_model
def test_model_1():

17
egs/aishell4/ASR/pruned_transducer_stateless5/train.py
View File

@ -53,20 +53,19 @@ from shutil import copyfile
from typing import Any, Dict, Optional, Tuple, Union
import k2
import optim
import torch
import torch.multiprocessing as mp
import torch.nn as nn
from asr_datamodule import Aishell4AsrDataModule
from conformer import Conformer
from decoder import Decoder
from joiner import Joiner
from .asr_datamodule import Aishell4AsrDataModule
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 local.text_normalize import text_normalize
from model import Transducer
from optim import Eden, Eve
from ..local.text_normalize import text_normalize
from .model import Transducer
from .optim import Eden, Eve, LRScheduler
from torch import Tensor
from torch.cuda.amp import GradScaler
from torch.nn.parallel import DistributedDataParallel as DDP
@ -86,7 +85,7 @@ from icefall.lexicon import Lexicon
from icefall.utils import AttributeDict, MetricsTracker, setup_logger, str2bool
LRSchedulerType = Union[
torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler
torch.optim.lr_scheduler._LRScheduler, LRScheduler
]