-
Notifications
You must be signed in to change notification settings - Fork 0
/
dependency_modeling.sh
executable file
·32 lines (28 loc) · 1.1 KB
/
dependency_modeling.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
#!/bin/bash
export DO_LOWER_CASE=False
export CACHE_DIR=../cache
export MAX_CONTEXT_SIZE=249
export MAX_SEQUENCE_LENGTH=512
export SPAN_LENGTH=1
export MAX_SPAN_LENGTH=1
export MAX_NUM_EXAMPLES=3000
#model_name_or_path=('bert-base-cased', 'bert-large-cased', 'roberta-base', 'roberta-large', 'xlnet-base-cased', 'xlnet-large-cased', 'xlm-mlm-en-2048')
#model_name=('bert', 'bert', 'roberta', 'roberta', 'xlnet', 'xlnet', 'xlm')
#data_name=('RACE', 'aclImdb')
export MODEL_NAME_OR_PATH=$1
export MODEL_NAME=$2 # bert, xlnet, xlm, roberta
export DATA_NAME=$3 # RACE, aclImdb
export OUTPUT_DIR=../output_dir/${MODEL_NAME_OR_PATH}-${DATA_NAME}
python dependency_modeling.py \
--data_name ${DATA_NAME} \
--model_type ${MODEL_NAME} \
--data_dir ../data/${DATA_NAME}/train \
--model_name_or_path ${MODEL_NAME_OR_PATH} \
--do_lower_case ${DO_LOWER_CASE} \
--max_seq_length ${MAX_SEQUENCE_LENGTH} \
--max_context_size ${MAX_CONTEXT_SIZE} \
--max_span_length ${MAX_SPAN_LENGTH} \
--span_length ${SPAN_LENGTH} \
--max_num_examples ${MAX_NUM_EXAMPLES} \
--output_dir ${OUTPUT_DIR} \
--cache_dir ${CACHE_DIR}