Skip to content

Commit

Permalink
bugfix: specify trainer.strategy=auto when devices=1 (NVIDIA#7509) (N…
Browse files Browse the repository at this point in the history
…VIDIA#7512)

Signed-off-by: Xuesong Yang <16880-xueyang@users.noreply.gitlab-master.nvidia.com>
Co-authored-by: Xuesong Yang <1646669+XuesongYang@users.noreply.github.com>
Co-authored-by: Xuesong Yang <16880-xueyang@users.noreply.gitlab-master.nvidia.com>
Signed-off-by: Sasha Meister <sasha.meister.work@gmail.com>
  • Loading branch information
3 people authored and ssh-meister committed Oct 10, 2023
1 parent 1c3c43c commit c2d70e6
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tutorials/tts/Vits_Training.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -309,6 +309,7 @@
" heteronyms_path=tts_dataset_files/heteronyms-052722 \\\n",
" trainer.max_epochs=3 \\\n",
" trainer.accelerator=auto \\\n",
" trainer.strategy=auto \\\n",
" trainer.check_val_every_n_epoch=1 \\\n",
" trainer.devices=1)"
]
Expand Down

0 comments on commit c2d70e6

Please sign in to comment.