Fix continue path

pull/1032/head
WeberJulian 2021-11-09 12:20:11 +01:00 committed by Eren Gölge
parent 61251bd86c
commit 5f40e96010
1 changed files with 2 additions and 1 deletions

View File

@ -9,6 +9,7 @@ from TTS.utils.audio import AudioProcessor
def main():
#os.environ["CUDA_VISIBLE_DEVICES"]=""
"""Run `tts` model training directly by a `config.json` file."""
# init trainer args
train_args = TrainingArgs()
@ -64,7 +65,7 @@ def main():
train_samples=train_samples,
eval_samples=eval_samples,
training_assets={"audio_processor": ap},
parse_command_line_args=False,
parse_command_line_args=True,
)
trainer.fit()