Merge remote-tracking branch 'origin/MassiveScaleToucan' into Massive…

…ScaleToucan
DigitalPhonetics · Oct 7, 2024 · 38ca97c · 38ca97c
2 parents 627579e + 2645299
commit 38ca97c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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,