mirror of https://github.com/coqui-ai/TTS.git
33 lines
904 B
Python
33 lines
904 B
Python
import os
|
|
import sys
|
|
import traceback
|
|
from TTS.utils.arguments import init_training
|
|
from TTS.utils.generic_utils import remove_experiment_folder
|
|
from TTS.trainer import TrainerTTS
|
|
|
|
|
|
def main():
|
|
# try:
|
|
args, config, OUT_PATH, AUDIO_PATH, c_logger, tb_logger = init_training(
|
|
sys.argv)
|
|
trainer = TrainerTTS(args,
|
|
config,
|
|
c_logger,
|
|
tb_logger,
|
|
output_path=OUT_PATH)
|
|
trainer.fit()
|
|
# except KeyboardInterrupt:
|
|
# remove_experiment_folder(OUT_PATH)
|
|
# try:
|
|
# sys.exit(0)
|
|
# except SystemExit:
|
|
# os._exit(0) # pylint: disable=protected-access
|
|
# except Exception: # pylint: disable=broad-except
|
|
# remove_experiment_folder(OUT_PATH)
|
|
# traceback.print_exc()
|
|
# sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|