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

remove output train folder at the end of the test

This commit is contained in:
Eren Gölge 2021-05-06 03:40:26 +02:00
parent 720fe13056
commit c6df8de80a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/tts_tests/test_speedy_speech_train.py
View File

@ -1,8 +1,8 @@
import glob
import os
import shutil
from tests import get_tests_output_path, run_cli
from TTS.config import BaseDatasetConfig
from TTS.tts.configs import SpeedySpeechConfig
config_path = os.path.join(get_tests_output_path(), "test_speedy_speech_config.json")
@ -46,3 +46,4 @@ continue_path = max(glob.glob(os.path.join(output_path, "*/")), key=os.path.getm
# restore the model and continue training for one more epoch
command_train = f"CUDA_VISIBLE_DEVICES='' python TTS/bin/train_speedy_speech.py --continue_path {continue_path} "
run_cli(command_train)
shutil.rmtree(continue_path)