chmanoj
/

xls-r-2B-te

Automatic Speech Recognition

Generated from Trainer

robust-speech-event

hf-asr-leaderboard

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

chmanoj commited on Jan 30, 2022

Commit

d1bf55f

•

1 Parent(s): 8ee27c4

Stop training

Files changed (1) hide show

run_bnb.sh +3 -4

run_bnb.sh CHANGED Viewed

@@ -9,7 +9,7 @@ python run_speech_recognition_ctc_bnb.py \
 --per_device_train_batch_size="4" \
 --per_device_eval_batch_size="4" \
 --gradient_accumulation_steps="8" \
---learning_rate="1e-6" \
 --warmup_steps="2000" \
 --length_column_name="input_length" \
 --evaluation_strategy="steps" \
@@ -23,7 +23,7 @@ python run_speech_recognition_ctc_bnb.py \
 --save_total_limit="2" \
 --freeze_feature_encoder \
 --feat_proj_dropout="0.0" \
---mask_time_prob="0.4" \
 --mask_time_length="10" \
 --mask_feature_prob="0.25" \
 --mask_feature_length="64" \
@@ -31,5 +31,4 @@ python run_speech_recognition_ctc_bnb.py \
 --use_auth_token \
 --fp16 \
 --group_by_length \
---do_train --do_eval \
---push_to_hub

 --per_device_train_batch_size="4" \
 --per_device_eval_batch_size="4" \
 --gradient_accumulation_steps="8" \
+--learning_rate="3e-6" \
 --warmup_steps="2000" \
 --length_column_name="input_length" \
 --evaluation_strategy="steps" \
 --save_total_limit="2" \
 --freeze_feature_encoder \
 --feat_proj_dropout="0.0" \
+--mask_time_prob="0.75" \
 --mask_time_length="10" \
 --mask_feature_prob="0.25" \
 --mask_feature_length="64" \
 --use_auth_token \
 --fp16 \
 --group_by_length \
+--do_train --do_eval