diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py index 9d63cb123..f31b3ab7b 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py @@ -21,8 +21,8 @@ import warnings from typing import List, Optional, Tuple import torch -from encoder_interface import EncoderInterface -from scaling import ( +from .encoder_interface import EncoderInterface +from .scaling import ( ActivationBalancer, BasicNorm, DoubleSwish, diff --git a/egs/librispeech/ASR/transducer_stateless/beam_search.py b/egs/librispeech/ASR/transducer_stateless/beam_search.py index ea985f30d..a33e6bc39 100644 --- a/egs/librispeech/ASR/transducer_stateless/beam_search.py +++ b/egs/librispeech/ASR/transducer_stateless/beam_search.py @@ -20,7 +20,7 @@ from typing import Dict, List, Optional import k2 import torch -from model import Transducer +from .model import Transducer from icefall.decode import Nbest, one_best_decoding from icefall.utils import get_texts diff --git a/egs/librispeech/ASR/transducer_stateless/conformer.py b/egs/librispeech/ASR/transducer_stateless/conformer.py index 2bf633201..0a4c11240 100644 --- a/egs/librispeech/ASR/transducer_stateless/conformer.py +++ b/egs/librispeech/ASR/transducer_stateless/conformer.py @@ -22,7 +22,7 @@ from typing import List, Optional, Tuple import torch from torch import Tensor, nn -from transformer import Transformer +from .transformer import Transformer from icefall.utils import make_pad_mask, subsequent_chunk_mask diff --git a/egs/librispeech/ASR/transducer_stateless/model.py b/egs/librispeech/ASR/transducer_stateless/model.py index 8281e1fb5..41cdece0f 100644 --- a/egs/librispeech/ASR/transducer_stateless/model.py +++ b/egs/librispeech/ASR/transducer_stateless/model.py @@ -19,7 +19,7 @@ import random import k2 import torch import torch.nn as nn -from encoder_interface import EncoderInterface +from .encoder_interface import EncoderInterface from icefall.utils import add_sos diff --git a/egs/librispeech/ASR/transducer_stateless/transformer.py b/egs/librispeech/ASR/transducer_stateless/transformer.py index e851dcc32..fe0ed49f8 100644 --- a/egs/librispeech/ASR/transducer_stateless/transformer.py +++ b/egs/librispeech/ASR/transducer_stateless/transformer.py @@ -20,8 +20,8 @@ from typing import Optional, Tuple import torch import torch.nn as nn -from encoder_interface import EncoderInterface -from subsampling import Conv2dSubsampling, VggSubsampling +from .encoder_interface import EncoderInterface +from .subsampling import Conv2dSubsampling, VggSubsampling from icefall.utils import make_pad_mask