diff --git a/Recipes/ToucanTTS_Nancy.py b/Recipes/ToucanTTS_Nancy.py index 57e97335..b0e196ac 100644 --- a/Recipes/ToucanTTS_Nancy.py +++ b/Recipes/ToucanTTS_Nancy.py @@ -65,6 +65,9 @@ def run(gpu_id, resume_checkpoint, finetune, model_dir, resume, use_wandb, wandb train_loop(net=model, datasets=[train_set], device=device, + warmup_steps=4000, + steps=200000, + batch_size=16, save_directory=save_dir, eval_lang="eng", path_to_checkpoint=resume_checkpoint,