From 10846568b97d15f1830d1f8c32a742a19b7bd615 Mon Sep 17 00:00:00 2001 From: shaynemei Date: Mon, 1 Aug 2022 21:39:21 -0700 Subject: [PATCH] use explicit relative imports for librispeech --- egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py | 4 ++-- egs/librispeech/ASR/transducer_stateless/beam_search.py | 2 +- egs/librispeech/ASR/transducer_stateless/conformer.py | 2 +- egs/librispeech/ASR/transducer_stateless/model.py | 2 +- egs/librispeech/ASR/transducer_stateless/transformer.py | 4 ++-- 5 files changed, 7 insertions(+), 7 deletions(-) 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