Skip to content

Commit 9dd9fcb

Browse files
alancuckinv-kkudrynski
authored andcommitted
[wav2vec2.0/PyT] Fix pip dependencies (librosa - numpy)
1 parent 4fca54f commit 9dd9fcb

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
editdistance==0.6.0
2-
librosa==0.8.0
2+
librosa==0.10.1
33
omegaconf==2.0.6 # optional for handling certain Fairseq ckpts
44
pyarrow==6.0.1
5-
soundfile==0.10.3.post1
5+
soundfile==0.12.1
66
sox==1.4.1
77
tqdm==4.53.0
88
git+https://github.com/NVIDIA/dllogger@v1.0.0#egg=dllogger

PyTorch/SpeechRecognition/wav2vec2/wav2vec2/arg_parser.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def populate_common(parser):
4848
help="Run for a number of epochs and exit")
4949
train.add_argument("--cudnn_benchmark", action="store_true",
5050
help="Enable cudnn benchmark")
51-
train.add_argument("--local_rank", default=os.getenv("LOCAL_RANK", 0),
51+
train.add_argument("--local_rank", "--local-rank", default=os.getenv("LOCAL_RANK", 0),
5252
type=int, help="GPU id used for distributed training")
5353

5454
optim = parser.add_argument_group("optimization setup")

0 commit comments

Comments
 (0)