diff --git a/TTS/bin/distribute_tts.py b/TTS/bin/distribute_tts.py index 1f94b62d..121acd5d 100644 --- a/TTS/bin/distribute_tts.py +++ b/TTS/bin/distribute_tts.py @@ -1,6 +1,14 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- +import os, sys +import pathlib +import time +import subprocess +import argparse +import torch + + def main(): """ Call train.py as a new process and pass command arguments @@ -25,15 +33,12 @@ def main(): ) args = parser.parse_args() - # OUT_PATH = create_experiment_folder(CONFIG.output_path, CONFIG.run_name, - # True) - # stdout_path = os.path.join(OUT_PATH, "process_stdout/") - num_gpus = torch.cuda.device_count() group_id = time.strftime("%Y_%m_%d-%H%M%S") # set arguments for train.py - command = ['train.py'] + folder_path =pathlib.Path(__file__).parent.absolute() + command = [os.path.join(folder_path,'train_tts.py')] command.append('--continue_path={}'.format(args.continue_path)) command.append('--restore_path={}'.format(args.restore_path)) command.append('--config_path={}'.format(args.config_path))