File size: 894 Bytes
11c11f8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | from __future__ import annotations
import argparse
from . import __version__
from .cli import infer_main, train_fast_main, train_hyper_main, train_main
def main() -> None:
parser = argparse.ArgumentParser(prog="python -m chimera")
parser.add_argument("--version", action="version", version=f"%(prog)s {__version__}")
subparsers = parser.add_subparsers(dest="command")
subparsers.add_parser("train")
subparsers.add_parser("train-fast")
subparsers.add_parser("train-hyper")
subparsers.add_parser("infer")
args, _ = parser.parse_known_args()
if args.command == "train":
train_main()
return
if args.command == "train-fast":
train_fast_main()
return
if args.command == "train-hyper":
train_hyper_main()
return
if args.command == "infer":
infer_main()
return
parser.print_help()
|