Replace gpus with devices (NVIDIA#7743)

Signed-off-by: Abhishree <abhishreetm@gmail.com>
bene-ges · Oct 17, 2023 · 8488496 · 8488496
1 parent 3f31216
commit 8488496
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/asr/experimental/k2/align_speech_parallel.py b/examples/asr/experimental/k2/align_speech_parallel.py
@@ -101,7 +101,7 @@ class ParallelAlignmentConfig:
  output_path: str = MISSING
  model_stride: int = 8
 
- trainer: TrainerConfig = field(default_factory=lambda: TrainerConfig(gpus=-1, accelerator="ddp"))
+ trainer: TrainerConfig = field(default_factory=lambda: TrainerConfig(devices=-1, accelerator="ddp"))
 
  # there arguments will be ignored
  return_predictions: bool = False

diff --git a/scripts/nemo_legacy_import/nlp_checkpoint_port.py b/scripts/nemo_legacy_import/nlp_checkpoint_port.py
@@ -81,7 +81,7 @@ def nemo_convert(argv):
 
  # Create a PL trainer object which is required for restoring Megatron models
  cfg_trainer = TrainerConfig(
- gpus=1,
+ devices=1,
  accelerator="ddp",
  num_nodes=1,
  # Need to set the following two to False as ExpManager will take care of them differently.