patrickvonplaten commited on
Commit
b4bf856
1 Parent(s): edf6414
Files changed (1) hide show
  1. run.sh +4 -4
run.sh CHANGED
@@ -1,7 +1,8 @@
1
  #!/usr/bin/env bash
2
- CUDA_VISIBLE_DEVICES="1" python run_speech_recognition_ctc.py \
 
3
  --dataset_name="timit_asr" \
4
- --model_name_or_path="anton-l/distilhubert" \
5
  --overwrite_output_dir \
6
  --output_dir="./distilhubert-timit" \
7
  --train_split_name="train" \
@@ -23,5 +24,4 @@ CUDA_VISIBLE_DEVICES="1" python run_speech_recognition_ctc.py \
23
  --fp16 \
24
  --group_by_length \
25
  --push_to_hub \
26
- --do_train --do_eval \
27
- --use_auth_token
 
1
  #!/usr/bin/env bash
2
+ # --gradient_checkpointing \
3
+ CUDA_VISIBLE_DEVICES="0" python run_speech_recognition_ctc.py \
4
  --dataset_name="timit_asr" \
5
+ --model_name_or_path="ntu-spml/distilhubert" \
6
  --overwrite_output_dir \
7
  --output_dir="./distilhubert-timit" \
8
  --train_split_name="train" \
 
24
  --fp16 \
25
  --group_by_length \
26
  --push_to_hub \
27
+ --do_train --do_eval