update strategy (NVIDIA#7577) (NVIDIA#7578)

Signed-off-by: Nithin Rao Koluguri <nithinraok> Co-authored-by: Nithin Rao <[email protected]> Signed-off-by: Sasha Meister <[email protected]>
ssh-meister · Oct 10, 2023 · 8406b69 · 8406b69
1 parent 468e8b0
commit 8406b69
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/asr/Self_Supervised_Pre_Training.ipynb b/tutorials/asr/Self_Supervised_Pre_Training.ipynb
@@ -312,7 +312,7 @@
         "\n",
         "if torch.cuda.is_available():\n",
         "  cfg.trainer.accelerator = 'gpu'\n",
-        "  cfg.trainer.strategy = 'dp'\n",
+        "  cfg.trainer.strategy = 'auto'\n",
         "  cfg.trainer.devices = 1\n",
         "else:\n",
         "  cfg.trainer.accelerator = 'cpu'\n",
@@ -534,7 +534,7 @@
         "\n",
         "if torch.cuda.is_available():\n",
         "  cfg.trainer.accelerator = 'gpu'\n",
-        "  cfg.trainer.strategy = 'dp'\n",
+        "  cfg.trainer.strategy = 'auto'\n",
         "  cfg.trainer.devices = 1\n",
         "else:\n",
         "  cfg.trainer.accelerator = 'cpu'\n",