minor fixes

This commit is contained in:
marcoyang1998 2023-07-18 11:14:06 +08:00
parent 0e7df7c5c4
commit 6939b3d6aa
2 changed files with 5 additions and 6 deletions

View File

@ -34,4 +34,4 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
main() main()

View File

@ -40,9 +40,8 @@ from lhotse import (
KaldifeatFbankConfig, KaldifeatFbankConfig,
LilcomChunkyWriter, LilcomChunkyWriter,
) )
from lhotse.recipes.utils import read_manifests_if_cached
from icefall.utils import get_executor, str2bool from icefall.utils import get_executor
# Torch's multithreaded behavior needs to be disabled or # Torch's multithreaded behavior needs to be disabled or
# it wastes a lot of CPU and slow things down. # it wastes a lot of CPU and slow things down.
@ -61,7 +60,7 @@ def get_args():
help="""Path to the bpe.model. If not None, we will remove short and help="""Path to the bpe.model. If not None, we will remove short and
long utterances before extracting features""", long utterances before extracting features""",
) )
parser.add_argument( parser.add_argument(
"--fbank-dir", "--fbank-dir",
type=str, type=str,
@ -102,14 +101,14 @@ def get_args():
"--start", "--start",
type=int, type=int,
default=0, default=0,
help="Process pieces starting from this number (inclusive). Only used in medium and large subset", help="Process pieces starting from this number (inclusive).",
) )
parser.add_argument( parser.add_argument(
"--stop", "--stop",
type=int, type=int,
default=-1, default=-1,
help="Stop processing pieces until this number (exclusive). Only used in medium and large subset", help="Stop processing pieces until this number (exclusive).",
) )
return parser.parse_args() return parser.parse_args()