From 4e79b8bc41dcb388bfc1072c14ad8d34b5a94201 Mon Sep 17 00:00:00 2001 From: erogol Date: Fri, 17 Jul 2020 13:23:25 +0200 Subject: [PATCH] fix distribute_tts --- TTS/bin/distribute_tts.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) 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))