wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-11 12:00:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactoring tests after Coqpit

Browse Source
This commit is contained in:
Eren Gölge
2021-05-11 11:30:00 +02:00
parent 87384c6008
commit 5aee30443f
29 changed files with 156 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/vocoder_tests/test_parallel_wavegan_train.py
+4 -3
View File
@@ -2,7 +2,8 @@ import glob
import os
import shutil
from tests import get_tests_output_path, run_cli
from tests import get_tests_output_path, run_cli, get_device_id
from TTS.vocoder.configs import ParallelWaveganConfig
config_path = os.path.join(get_tests_output_path(), "test_vocoder_config.json")
@@ -28,13 +29,13 @@ config.audio.trim_db = 60
config.save_json(config_path)
# train the model for one epoch
command_train = f"CUDA_VISIBLE_DEVICES='' python TTS/bin/train_vocoder_gan.py --config_path {config_path} "
command_train = f"CUDA_VISIBLE_DEVICES='{get_device_id()}' python TTS/bin/train_vocoder_gan.py --config_path {config_path} "
run_cli(command_train)
# Find latest folder
continue_path = max(glob.glob(os.path.join(output_path, "*/")), key=os.path.getmtime)
# restore the model and continue training for one more epoch
command_train = f"CUDA_VISIBLE_DEVICES='' python TTS/bin/train_vocoder_gan.py --continue_path {continue_path} "
command_train = f"CUDA_VISIBLE_DEVICES='{get_device_id()}' python TTS/bin/train_vocoder_gan.py --continue_path {continue_path} "
run_cli(command_train)
shutil.rmtree(continue_path)