[TTS] fixed trainer's accelerator and strategy. (#7569)

Signed-off-by: Xuesong Yang <[email protected]> Co-authored-by: Xuesong Yang <[email protected]>
NVIDIA · Sep 29, 2023 · 923fc0f · 923fc0f
1 parent eccf98f
commit 923fc0f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/tts/Vits_Training.ipynb b/tutorials/tts/Vits_Training.ipynb
@@ -304,8 +304,8 @@
     "  phoneme_dict_path=tts_dataset_files/ipa_cmudict-0.7b_nv23.01.txt \\\n",
     "  heteronyms_path=tts_dataset_files/heteronyms-052722 \\\n",
     "  trainer.max_epochs=3 \\\n",
-    "  trainer.accelerator=auto \\\n",
-    "  trainer.strategy=auto \\\n",
+    "  trainer.accelerator='gpu' \\\n",
+    "  trainer.strategy='ddp_find_unused_parameters_true' \\\n",
     "  trainer.check_val_every_n_epoch=1 \\\n",
     "  trainer.devices=1)"
    ]