Commit
•
b4bf856
1
Parent(s):
edf6414
up
Browse files
run.sh
CHANGED
|
@@ -1,7 +1,8 @@
|
|
| 1 |
#!/usr/bin/env bash
|
| 2 |
-
|
|
|
|
| 3 |
--dataset_name="timit_asr" \
|
| 4 |
-
--model_name_or_path="
|
| 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
|
|
|