eyna commited on
Commit
a83fcbd
1 Parent(s): d5d1dd6

Create train.sh

Browse files
Files changed (1) hide show
  1. train.sh +19 -0
train.sh ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ./run_wav2vec2_pretrain_flax.py \
2
+ --output_dir="./output50" \
3
+ --speech_file_column="path" \
4
+ --num_train_epochs="50" \
5
+ --per_device_train_batch_size="32" \
6
+ --per_device_eval_batch_size="32" \
7
+ --validation_split_percentage="5" \
8
+ --learning_rate="5e-4" \
9
+ --weight_decay="0.01" \
10
+ --warmup_steps="1000" \
11
+ --model_name_or_path="./" \
12
+ --dataset_name="common_voice" \
13
+ --dataset_config_name="dv" \
14
+ --preprocessing_num_workers="96" \
15
+ --max_duration_in_seconds="10.0" \
16
+ --adam_beta1="0.9" \
17
+ --adam_beta2="0.98" \
18
+ --dtype="bfloat16" \
19
+ --pad_to_multiple_of="16384"