Fix continue path

This commit is contained in:
WeberJulian
2021-12-20 11:54:10 +00:00
committed by Eren Gölge
parent 120332d53f
commit 23d789c072
+2 -1
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()