Source code for espnet2.bin.pack

#!/usr/bin/env python3
import argparse
from typing import Type

from espnet2.main_funcs.pack_funcs import pack


[docs]class PackedContents: files = [] yaml_files = []
[docs]class ASRPackedContents(PackedContents): # These names must be consistent with the argument of inference functions files = ["asr_model_file", "lm_file"] yaml_files = ["asr_train_config", "lm_train_config"]
[docs]class STPackedContents(PackedContents): files = ["st_model_file"] yaml_files = ["st_train_config"]
[docs]class TTSPackedContents(PackedContents): files = ["model_file"] yaml_files = ["train_config"]
[docs]class EnhPackedContents(PackedContents): files = ["model_file"] yaml_files = ["train_config"]
[docs]class DiarPackedContents(PackedContents): files = ["model_file"] yaml_files = ["train_config"]
[docs]class SVSPackedContents(PackedContents): files = ["model_file"] yaml_files = ["train_config"]
[docs]class EnhS2TPackedContents(PackedContents): # These names must be consistent with the argument of inference functions files = ["enh_s2t_model_file", "lm_file"] yaml_files = ["enh_s2t_train_config", "lm_train_config"]
[docs]class SSLPackedContents(PackedContents): # These names must be consistent with the argument of inference functions files = ["model_file"] yaml_files = ["train_config"]
[docs]def add_arguments(parser: argparse.ArgumentParser, contents: Type[PackedContents]): parser.add_argument("--outpath", type=str, required=True) for key in contents.yaml_files: parser.add_argument(f"--{key}", type=str, default=None) for key in contents.files: parser.add_argument(f"--{key}", type=str, default=None) parser.add_argument("--option", type=str, action="append", default=[])
[docs]def get_parser() -> argparse.ArgumentParser: parser = argparse.ArgumentParser(description="Pack input files to archive format") subparsers = parser.add_subparsers() # Create subparser for ASR for name, contents in [ ("asr", ASRPackedContents), ("st", STPackedContents), ("tts", TTSPackedContents), ("enh", EnhPackedContents), ("diar", DiarPackedContents), ("svs", SVSPackedContents), ("enh_s2t", EnhS2TPackedContents), ("ssl", SSLPackedContents), ]: parser_asr = subparsers.add_parser( name, formatter_class=argparse.ArgumentDefaultsHelpFormatter, ) add_arguments(parser_asr, contents) parser_asr.set_defaults(contents=contents) return parser
[docs]def main(cmd=None): parser = get_parser() args = parser.parse_args(cmd) if not hasattr(args, "contents"): parser.print_help() parser.exit(2) yaml_files = { y: getattr(args, y) for y in args.contents.yaml_files if getattr(args, y) is not None } files = { y: getattr(args, y) for y in args.contents.files if getattr(args, y) is not None } pack( yaml_files=yaml_files, files=files, option=args.option, outpath=args.outpath, )
if __name__ == "__main__": main()