m3hrdadfi commited on
Commit
44da8cd
1 Parent(s): 6a6f772

Fix script

Browse files
src/run_persian.sh CHANGED
@@ -8,7 +8,7 @@ export OUTPUT_DIR=/home/m3hrdadfi/data_cache/
8
  export MODEL_NAME_OR_PATH=/home/m3hrdadfi/code/wav2vec2-base-persian
9
 
10
 
11
- export TRAIN_FILE=/home/m3hrdadfi/data/fa/train.csv
12
  export VALIDATION_FILE=/home/m3hrdadfi/data/fa/test.csv
13
  export SPEECH_FILE_COLUMN=path
14
 
 
8
  export MODEL_NAME_OR_PATH=/home/m3hrdadfi/code/wav2vec2-base-persian
9
 
10
 
11
+ export TRAIN_FILE=/home/m3hrdadfi/data/fa/train_with_aug.csv
12
  export VALIDATION_FILE=/home/m3hrdadfi/data/fa/test.csv
13
  export SPEECH_FILE_COLUMN=path
14
 
src/run_wav2vec2_pretrain_flax.py CHANGED
@@ -47,8 +47,6 @@ from transformers import (
47
  )
48
  from transformers.models.wav2vec2.modeling_flax_wav2vec2 import _compute_mask_indices, _sample_negative_indices
49
 
50
- from normalizer import normalizer
51
-
52
  logger = logging.getLogger(__name__)
53
 
54
 
 
47
  )
48
  from transformers.models.wav2vec2.modeling_flax_wav2vec2 import _compute_mask_indices, _sample_negative_indices
49
 
 
 
50
  logger = logging.getLogger(__name__)
51
 
52