patrickvonplaten
/

distilhubert-timit

Automatic Speech Recognition Transformers PyTorch TensorBoard

hubert timit_asr generated_from_trainer Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

patrickvonplaten commited on Oct 28, 2021

Commit

b4bf856

•

1 Parent(s): edf6414

up

Files changed (1) hide show

run.sh +4 -4

run.sh CHANGED Viewed

@@ -1,7 +1,8 @@
 #!/usr/bin/env bash
-CUDA_VISIBLE_DEVICES="1" python run_speech_recognition_ctc.py \
 	--dataset_name="timit_asr" \
-	--model_name_or_path="anton-l/distilhubert" \
 	--overwrite_output_dir \
 	--output_dir="./distilhubert-timit" \
 	--train_split_name="train" \
@@ -23,5 +24,4 @@ CUDA_VISIBLE_DEVICES="1" python run_speech_recognition_ctc.py \
 	--fp16 \
 	--group_by_length \
 	--push_to_hub \
-	--do_train --do_eval \
-	--use_auth_token

 #!/usr/bin/env bash
+#	--gradient_checkpointing \
+CUDA_VISIBLE_DEVICES="0" python run_speech_recognition_ctc.py \
 	--dataset_name="timit_asr" \
+	--model_name_or_path="ntu-spml/distilhubert" \
 	--overwrite_output_dir \
 	--output_dir="./distilhubert-timit" \
 	--train_split_name="train" \
 	--fp16 \
 	--group_by_length \
 	--push_to_hub \
+	--do_train --do_eval