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

Signed-off-by: Nithin Rao Koluguri <nithinraok> Co-authored-by: Nithin Rao <[email protected]> Signed-off-by: Elena Rastorgueva <[email protected]>
erastorgueva-nv · Oct 11, 2023 · f9ddc19 · f9ddc19
1 parent 1b53e91
commit f9ddc19
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",