salti commited on
Commit
ed1c4e0
1 Parent(s): 8d9b681

add running script

Browse files
Files changed (1) hide show
  1. run-t5v1_1-small.sh +27 -0
run-t5v1_1-small.sh ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ export model_dir=arabic-t5-small
2
+ export train_batch_size=48
3
+ export eval_batch_size=48
4
+
5
+ python ./run_t5_mlm_flax.py \
6
+ --model_type t5 \
7
+ --config_name ${model_dir} \
8
+ --tokenizer_name ${model_dir} \
9
+ --use_fast_tokenizer True \
10
+ --dtype float32 \
11
+ --max_seq_length 512 \
12
+ --preprocessing_num_workers 96 \
13
+ --output_dir ${model_dir} \
14
+ --overwrite_output_dir True \
15
+ --do_train \
16
+ --per_device_train_batch_size ${train_batch_size} \
17
+ --per_device_eval_batch_size ${eval_batch_size} \
18
+ --learning_rate 1e-2 \
19
+ --num_train_epochs 1 \
20
+ --logging_steps 100 \
21
+ --eval_steps 2500 \
22
+ --save_steps 10000 \
23
+ --seed 12 \
24
+ --adafactor True \
25
+ --push_to_hub \
26
+ --cache_dir ./training_cache \
27
+ --save_total_limit 5