run.py

import argparse
import importlib

# Parse args
parser = argparse.ArgumentParser(
    description="Command to start PIT training, configured by .yaml files")
parser.add_argument(
    "--model",
    type=str,
    default="IQformer_v6_0_0",
    dest="model",
    help="Insert model name")
parser.add_argument(
    "--engine-mode",
    choices=["train", "test", "test_save", "infer_sample"],
    default="train",
    help="This option is used to chooose the mode")
parser.add_argument(
    "--sample-file",
    type=str,
    default=None,
    help="directoy for sample audio")
parser.add_argument(
    "--out-wav-dir",
    type=str,
    default=None,
    help="This option is used to specficy save directory for output wav file in test_wav mode")
args = parser.parse_args()

# Call target model
main_module = importlib.import_module(f"models.{args.model}.main")
main_module.main(args)