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 #412

This commit is contained in:
erogol 2020-05-25 10:32:14 +02:00
parent ac55f198f1
commit ef7e31c498
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
speaker_encoder/train.py
View File

@ -13,12 +13,11 @@ from TTS.speaker_encoder.model import SpeakerEncoder
from TTS.speaker_encoder.visual import plot_embeddings from TTS.speaker_encoder.visual import plot_embeddings
from TTS.speaker_encoder.generic_utils import save_best_model from TTS.speaker_encoder.generic_utils import save_best_model
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.utils.generic_utils import (NoamLR, check_update, copy_config_file, from TTS.utils.generic_utils import (create_experiment_folder, get_git_branch,
count_parameters,
create_experiment_folder, get_git_branch,
load_config,
remove_experiment_folder, set_init_dict) remove_experiment_folder, set_init_dict)
from TTS.utils.logger import Logger from TTS.utils.io import load_config, copy_config_file
from TTS.utils.training import check_update, NoamLR
from TTS.utils.tensorboard_logger import TensorboardLogger
from TTS.utils.radam import RAdam from TTS.utils.radam import RAdam
torch.backends.cudnn.enabled = True torch.backends.cudnn.enabled = True
@ -237,7 +236,7 @@ if __name__ == '__main__':
new_fields) new_fields)
LOG_DIR = OUT_PATH LOG_DIR = OUT_PATH
tb_logger = Logger(LOG_DIR) tb_logger = TensorboardLogger(LOG_DIR)
try: try:
main(args) main(args)