This repository has been archived by the owner on Jan 1, 2025. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathconfig.yaml
210 lines (210 loc) · 4.41 KB
/
config.yaml
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
data_path: anli/data
outp_path: data/
eval_data: mnli_m_dev
batch_size: 128
hyp_distinct_perm: True
prem_distinct_perm: True
load_model: False
rand_data_folder: '' # override folder_name
data_prep_config:
num_tries: 100
percent: 1.0
keep_order: 0.0
retain_stop: False
retain_punct: True
save_data: True
rebuild: False
keep_premise: False
lang: "en"
mnli_train:
orig_path: build/mnli/train.jsonl
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
mnli_m_dev:
orig_path: build/mnli/m_dev.jsonl
sent1_label: premise
sent2_label: hypothesis
index_label: uid
target_label: label
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
mnli_mm_dev:
orig_path: build/mnli/mm_dev.jsonl
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
snli_dev:
orig_path: build/snli/dev.jsonl
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
snli_test:
orig_path: build/snli/test.jsonl
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
anli_r1_dev:
orig_path: build/anli/r1/dev.jsonl
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
anli_r2_dev:
orig_path: build/anli/r2/dev.jsonl
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
anli_r3_dev:
orig_path: build/anli/r3/dev.jsonl
id_map:
entailment_id: 2
contradiction_id: 0
neutral_id: 1
rte_dev:
orig_path: lib/fairseq/glue_data/RTE/processed
sent1_label: sentence1
sent2_label: sentence2
index_label: index
target_label: target
qqp_dev:
orig_path: lib/fairseq/glue_data/QQP/processed
sent1_label: sentence1
sent2_label: sentence2
index_label: index
target_label: target
qnli_dev:
orig_path: lib/fairseq/glue_data/QNLI/processed
sent1_label: sentence1
sent2_label: sentence2
index_label: index
target_label: target
ocnli_dev:
orig_path: build/ocnli/dev.jsonl
model_type: hub
hub:
git_repo: pytorch/fairseq
model_name: roberta.large.mnli
hf_mnli_distilbert:
train_data: mnli
train_mode: orig
model_name: distilbert
id_map:
entailment_id: 0
contradiction_id: 2
neutral_id: 1
hf_ocnli_roberta:
train_data: ocnli
train_mode: orig
model_name: chinese-roberta-large
id_map:
entailment_id: 0
contradiction_id: 2
neutral_id: 1
rnn_infersent:
model_name: infersent
encoder_type: InferSent
train_data: ocnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_convnet:
model_name: convnet
encoder_type: ConvNetEncoder
train_data: ocnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_bilstm:
model_name: bilstm
encoder_type: BLSTMprojEncoder
train_data: mnli
rnn_blstmprojencoder:
model_name: blstmprojencoder
encoder_type: BLSTMprojEncoder
train_data: mnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_bgrulastencoder:
model_name: bgrulastencoder
encoder_type: BGRUlastEncoder
train_data: ocnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_innerattentionmila:
model_name: innerattentionmilaencoder
encoder_type: InnerAttentionMILAEncoder
train_data: ocnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_innerattentionnaacl:
model_name: innerattentionnaaclencoder
encoder_type: InnerAttentionNAACLEncoder
train_data: ocnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_lstmencoder:
model_name: lstmencoder
encoder_type: LSTMEncoder
train_data: ocnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_infersent_rand:
outputdir: savedir_mnli_rand
nlipath: mnli_rand
model_name: infersent
encoder_type: InferSent
train_data: mnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_convnet_rand:
outputdir: savedir_mnli_rand
nlipath: mnli_rand
model_name: convnet
encoder_type: ConvNetEncoder
train_data: mnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2
rnn_blstmprojencoder_rand:
outputdir: savedir_mnli_rand
nlipath: mnli_rand
model_name: blstmprojencoder
encoder_type: BLSTMprojEncoder
train_data: mnli
train_mode: orig
id_map:
entailment_id: 0
contradiction_id: 1
neutral_id: 2