gpus -> devices (#7542)

Signed-off-by: Nithin Rao Koluguri <nithinraok> Co-authored-by: Nithin Rao Koluguri <nithinraok>
NVIDIA · Sep 27, 2023 · 61881ae · 61881ae
1 parent 7ded0c6
commit 61881ae
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tutorials/asr/Self_Supervised_Pre_Training.ipynb b/tutorials/asr/Self_Supervised_Pre_Training.ipynb
@@ -313,11 +313,11 @@
         "if torch.cuda.is_available():\n",
         "  cfg.trainer.accelerator = 'gpu'\n",
         "  cfg.trainer.strategy = 'dp'\n",
-        "  cfg.trainer.gpus = 1\n",
+        "  cfg.trainer.devices = 1\n",
         "else:\n",
         "  cfg.trainer.accelerator = 'cpu'\n",
         "  cfg.trainer.strategy = auto\n",
-        "  cfg.trainer.gpus = 0\n",
+        "  cfg.trainer.devices = 0\n",
         "\n",
         "cfg.exp_manager.exp_dir = data_dir + \"/content/exp\"\n",
         "cfg.exp_manager.name = \"pre_trained\"\n",
@@ -535,11 +535,11 @@
         "if torch.cuda.is_available():\n",
         "  cfg.trainer.accelerator = 'gpu'\n",
         "  cfg.trainer.strategy = 'dp'\n",
-        "  cfg.trainer.gpus = 1\n",
+        "  cfg.trainer.devices = 1\n",
         "else:\n",
         "  cfg.trainer.accelerator = 'cpu'\n",
         "  cfg.trainer.strategy = auto\n",
-        "  cfg.trainer.gpus = 0\n",
+        "  cfg.trainer.devices = 0\n",
         "\n",
         "cfg.model.tokenizer.dir = data_dir + \"/tokenizers/an4/tokenizer_spe_unigram_v128/\"  # note this is a directory, not a path to a vocabulary file\n",
         "cfg.model.tokenizer.type = \"bpe\"\n",