Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix distribute_tts

This commit is contained in:
erogol 2020-07-17 13:23:25 +02:00
parent a0f488136a
commit 4e79b8bc41
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
TTS/bin/distribute_tts.py
View File

@ -1,6 +1,14 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import os, sys
import pathlib
import time
import subprocess
import argparse
import torch
def main(): def main():
""" """
Call train.py as a new process and pass command arguments Call train.py as a new process and pass command arguments
@ -25,15 +33,12 @@ def main():
) )
args = parser.parse_args() 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() num_gpus = torch.cuda.device_count()
group_id = time.strftime("%Y_%m_%d-%H%M%S") group_id = time.strftime("%Y_%m_%d-%H%M%S")
# set arguments for train.py # 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('--continue_path={}'.format(args.continue_path))
command.append('--restore_path={}'.format(args.restore_path)) command.append('--restore_path={}'.format(args.restore_path))
command.append('--config_path={}'.format(args.config_path)) command.append('--config_path={}'.format(args.config_path))