Skip to content

Commit

Permalink
Revert trainer.gpus to trainer.devices in Self_Supervised_Pre_Trainin…
Browse files Browse the repository at this point in the history
…g.ipynb

Signed-off-by: Abhishree <abhishreetm@gmail.com>
  • Loading branch information
athitten committed Sep 27, 2023
1 parent 6eb6276 commit 1fdec9a
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions tutorials/asr/Self_Supervised_Pre_Training.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -316,7 +316,7 @@
" cfg.trainer.devices = 1\n",
"else:\n",
" cfg.trainer.accelerator = 'cpu'\n",
" cfg.trainer.strategy = auto\n",
" cfg.trainer.strategy = 'auto'\n",
" cfg.trainer.devices = 0\n",
"\n",
"cfg.exp_manager.exp_dir = data_dir + \"/content/exp\"\n",
Expand Down Expand Up @@ -538,13 +538,8 @@
" cfg.trainer.devices = 1\n",
"else:\n",
" cfg.trainer.accelerator = 'cpu'\n",
<<<<<<< HEAD
" cfg.trainer.strategy = auto\n",
" cfg.trainer.devices = 0\n",
=======
" cfg.trainer.strategy = 'auto'\n",
" cfg.trainer.gpus = 0\n",
>>>>>>> Add missing quotes for auto strategy
" 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",
Expand Down

0 comments on commit 1fdec9a

Please sign in to comment.