|
from pathlib import Path |
|
from toolbox import Toolbox |
|
from utils.argutils import print_args |
|
from utils.modelutils import check_model_paths |
|
import argparse |
|
import os |
|
|
|
|
|
if __name__ == '__main__': |
|
parser = argparse.ArgumentParser( |
|
description="Runs the toolbox", |
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter |
|
) |
|
|
|
parser.add_argument("-d", "--datasets_root", type=Path, help= \ |
|
"Path to the directory containing your datasets. See toolbox/__init__.py for a list of " |
|
"supported datasets.", default=None) |
|
parser.add_argument("-e", "--enc_models_dir", type=Path, default="encoder/saved_models", |
|
help="Directory containing saved encoder models") |
|
parser.add_argument("-s", "--syn_models_dir", type=Path, default="synthesizer/saved_models", |
|
help="Directory containing saved synthesizer models") |
|
parser.add_argument("-v", "--voc_models_dir", type=Path, default="vocoder/saved_models", |
|
help="Directory containing saved vocoder models") |
|
parser.add_argument("--cpu", action="store_true", help=\ |
|
"If True, processing is done on CPU, even when a GPU is available.") |
|
parser.add_argument("--seed", type=int, default=None, help=\ |
|
"Optional random number seed value to make toolbox deterministic.") |
|
parser.add_argument("--no_mp3_support", action="store_true", help=\ |
|
"If True, no mp3 files are allowed.") |
|
args = parser.parse_args() |
|
print_args(args, parser) |
|
|
|
if args.cpu: |
|
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = "" |
|
del args.cpu |
|
|
|
|
|
check_model_paths(encoder_path=args.enc_models_dir, synthesizer_path=args.syn_models_dir, |
|
vocoder_path=args.voc_models_dir) |
|
|
|
|
|
Toolbox(**vars(args)) |
|
|