[wav2vec2.0/PyT] Fix pip dependencies (librosa - numpy)

xvdp · Dec 8, 2023 · 9dd9fcb · 9dd9fcb
1 parent 4fca54f
commit 9dd9fcb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/PyTorch/SpeechRecognition/wav2vec2/requirements.txt b/PyTorch/SpeechRecognition/wav2vec2/requirements.txt
@@ -1,8 +1,8 @@
 editdistance==0.6.0
-librosa==0.8.0
+librosa==0.10.1
 omegaconf==2.0.6  # optional for handling certain Fairseq ckpts
 pyarrow==6.0.1
-soundfile==0.10.3.post1
+soundfile==0.12.1
 sox==1.4.1
 tqdm==4.53.0
 git+https://github.com/NVIDIA/dllogger@v1.0.0#egg=dllogger
diff --git a/PyTorch/SpeechRecognition/wav2vec2/wav2vec2/arg_parser.py b/PyTorch/SpeechRecognition/wav2vec2/wav2vec2/arg_parser.py
@@ -48,7 +48,7 @@ def populate_common(parser):
                        help="Run for a number of epochs and exit")
     train.add_argument("--cudnn_benchmark", action="store_true",
                        help="Enable cudnn benchmark")
-    train.add_argument("--local_rank", default=os.getenv("LOCAL_RANK", 0),
+    train.add_argument("--local_rank", "--local-rank", default=os.getenv("LOCAL_RANK", 0),
                        type=int, help="GPU id used for distributed training")
 
     optim = parser.add_argument_group("optimization setup")