Browse Source

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

Adrian Lancucki 2 năm trước cách đây
mục cha
commit
9dd9fcb98f

+ 2 - 2
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/[email protected]#egg=dllogger

+ 1 - 1
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")