CarelvNiekerk commited on
Commit
9030642
1 Parent(s): ab8f0f8

Upload config

Browse files
Files changed (1) hide show
  1. config.json +330 -0
config.json ADDED
@@ -0,0 +1,330 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "results/setsumbt/EnD2-SetSUMBT+ActPrediction-multiwoz21-roberta-gru-cosine-distribution_distillation-Seed0-30-03-23-19-34",
3
+ "active_domain_loss_weight": 0.2,
4
+ "architectures": [
5
+ "RobertaSetSUMBT"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.1,
8
+ "bos_token_id": 0,
9
+ "candidate_embedding_model_name": "roberta-base",
10
+ "classifier_dropout": null,
11
+ "code_version": "setsumbt/a4dd9b74f22d0ef72c9fe7bcf125eb61d3f5f4f7",
12
+ "dev_batch_size": 12,
13
+ "distance_measure": "cosine",
14
+ "domain_ids": {
15
+ "attraction": [
16
+ 1,
17
+ 2,
18
+ 3,
19
+ 4,
20
+ 5,
21
+ 6,
22
+ 7,
23
+ 8,
24
+ 9
25
+ ],
26
+ "hospital": [
27
+ 10,
28
+ 11,
29
+ 12,
30
+ 13
31
+ ],
32
+ "hotel": [
33
+ 14,
34
+ 15,
35
+ 16,
36
+ 17,
37
+ 18,
38
+ 19,
39
+ 20,
40
+ 21,
41
+ 22,
42
+ 23,
43
+ 24,
44
+ 25,
45
+ 26,
46
+ 27,
47
+ 28
48
+ ],
49
+ "police": [
50
+ 58,
51
+ 59,
52
+ 60,
53
+ 61
54
+ ],
55
+ "restaurant": [
56
+ 32,
57
+ 33,
58
+ 34,
59
+ 35,
60
+ 36,
61
+ 37,
62
+ 38,
63
+ 39,
64
+ 40,
65
+ 29,
66
+ 30,
67
+ 31
68
+ ],
69
+ "taxi": [
70
+ 41,
71
+ 42,
72
+ 43,
73
+ 44,
74
+ 45,
75
+ 46
76
+ ],
77
+ "train": [
78
+ 47,
79
+ 48,
80
+ 49,
81
+ 50,
82
+ 51,
83
+ 52,
84
+ 53,
85
+ 54,
86
+ 55,
87
+ 56,
88
+ 57
89
+ ]
90
+ },
91
+ "dropout_rate": 0.3,
92
+ "eos_token_id": 2,
93
+ "freeze_encoder": false,
94
+ "hidden_act": "gelu",
95
+ "hidden_dropout_prob": 0.1,
96
+ "hidden_size": 768,
97
+ "informable_slot_ids": {
98
+ "attraction-area": 2,
99
+ "attraction-name": 5,
100
+ "attraction-type": 9,
101
+ "hospital-department": 11,
102
+ "hotel-area": 15,
103
+ "hotel-book day": 16,
104
+ "hotel-book people": 17,
105
+ "hotel-book stay": 18,
106
+ "hotel-internet": 20,
107
+ "hotel-name": 21,
108
+ "hotel-parking": 22,
109
+ "hotel-price range": 25,
110
+ "hotel-stars": 27,
111
+ "hotel-type": 28,
112
+ "restaurant-area": 30,
113
+ "restaurant-book day": 31,
114
+ "restaurant-book people": 32,
115
+ "restaurant-book time": 33,
116
+ "restaurant-food": 35,
117
+ "restaurant-name": 36,
118
+ "restaurant-price range": 39,
119
+ "taxi-arrive by": 41,
120
+ "taxi-departure": 42,
121
+ "taxi-destination": 43,
122
+ "taxi-leave at": 44,
123
+ "train-arrive by": 47,
124
+ "train-book people": 48,
125
+ "train-day": 50,
126
+ "train-departure": 51,
127
+ "train-destination": 52,
128
+ "train-leave at": 54
129
+ },
130
+ "initializer_range": 0.02,
131
+ "intermediate_size": 3072,
132
+ "layer_norm_eps": 1e-05,
133
+ "loss_function": "distribution_distillation",
134
+ "max_candidate_len": 12,
135
+ "max_dialogue_len": 12,
136
+ "max_position_embeddings": 514,
137
+ "max_slot_len": 12,
138
+ "max_turn_len": 64,
139
+ "model_type": "roberta",
140
+ "nbt_hidden_size": 300,
141
+ "nbt_layers": 1,
142
+ "nbt_type": "gru",
143
+ "num_attention_heads": 12,
144
+ "num_hidden_layers": 12,
145
+ "num_slots": 62,
146
+ "num_values": {
147
+ "attraction-address": 1,
148
+ "attraction-area": 7,
149
+ "attraction-choice": 1,
150
+ "attraction-entrance fee": 1,
151
+ "attraction-name": 156,
152
+ "attraction-open hours": 1,
153
+ "attraction-phone": 1,
154
+ "attraction-postcode": 1,
155
+ "attraction-type": 14,
156
+ "hospital-address": 1,
157
+ "hospital-department": 49,
158
+ "hospital-phone": 1,
159
+ "hospital-postcode": 1,
160
+ "hotel-address": 1,
161
+ "hotel-area": 7,
162
+ "hotel-book day": 9,
163
+ "hotel-book people": 12,
164
+ "hotel-book stay": 12,
165
+ "hotel-choice": 1,
166
+ "hotel-internet": 5,
167
+ "hotel-name": 72,
168
+ "hotel-parking": 5,
169
+ "hotel-phone": 1,
170
+ "hotel-postcode": 1,
171
+ "hotel-price range": 5,
172
+ "hotel-ref": 1,
173
+ "hotel-stars": 8,
174
+ "hotel-type": 5,
175
+ "police-address": 1,
176
+ "police-name": 1,
177
+ "police-phone": 1,
178
+ "police-postcode": 1,
179
+ "restaurant-address": 1,
180
+ "restaurant-area": 7,
181
+ "restaurant-book day": 9,
182
+ "restaurant-book people": 12,
183
+ "restaurant-book time": 290,
184
+ "restaurant-choice": 1,
185
+ "restaurant-food": 103,
186
+ "restaurant-name": 185,
187
+ "restaurant-phone": 1,
188
+ "restaurant-postcode": 1,
189
+ "restaurant-price range": 5,
190
+ "restaurant-ref": 1,
191
+ "taxi-arrive by": 290,
192
+ "taxi-departure": 255,
193
+ "taxi-destination": 253,
194
+ "taxi-leave at": 290,
195
+ "taxi-phone": 1,
196
+ "taxi-type": 1,
197
+ "train-arrive by": 290,
198
+ "train-book people": 12,
199
+ "train-choice": 1,
200
+ "train-day": 9,
201
+ "train-departure": 32,
202
+ "train-destination": 28,
203
+ "train-duration": 1,
204
+ "train-leave at": 290,
205
+ "train-price": 1,
206
+ "train-ref": 1,
207
+ "train-train id": 1
208
+ },
209
+ "pad_token_id": 1,
210
+ "position_embedding_type": "absolute",
211
+ "predict_actions": true,
212
+ "requestable_slot_ids": {
213
+ "attraction-address": 1,
214
+ "attraction-area": 2,
215
+ "attraction-entrance fee": 4,
216
+ "attraction-phone": 7,
217
+ "attraction-postcode": 8,
218
+ "attraction-type": 9,
219
+ "hospital-address": 10,
220
+ "hospital-phone": 12,
221
+ "hospital-postcode": 13,
222
+ "hotel-address": 14,
223
+ "hotel-area": 15,
224
+ "hotel-internet": 20,
225
+ "hotel-parking": 22,
226
+ "hotel-phone": 23,
227
+ "hotel-postcode": 24,
228
+ "hotel-price range": 25,
229
+ "hotel-ref": 26,
230
+ "hotel-stars": 27,
231
+ "hotel-type": 28,
232
+ "police-address": 58,
233
+ "police-phone": 60,
234
+ "police-postcode": 61,
235
+ "restaurant-address": 29,
236
+ "restaurant-area": 30,
237
+ "restaurant-food": 35,
238
+ "restaurant-phone": 37,
239
+ "restaurant-postcode": 38,
240
+ "restaurant-price range": 39,
241
+ "restaurant-ref": 40,
242
+ "taxi-phone": 45,
243
+ "taxi-type": 46,
244
+ "train-arrive by": 47,
245
+ "train-duration": 53,
246
+ "train-leave at": 54,
247
+ "train-price": 55,
248
+ "train-ref": 56,
249
+ "train-train id": 57
250
+ },
251
+ "rnn_zero_init": false,
252
+ "seed": 0,
253
+ "set_pooling": "cnn",
254
+ "set_similarity": true,
255
+ "slot_attention_heads": 12,
256
+ "slot_ids": {
257
+ "attraction-address": 1,
258
+ "attraction-area": 2,
259
+ "attraction-choice": 3,
260
+ "attraction-entrance fee": 4,
261
+ "attraction-name": 5,
262
+ "attraction-open hours": 6,
263
+ "attraction-phone": 7,
264
+ "attraction-postcode": 8,
265
+ "attraction-type": 9,
266
+ "hospital-address": 10,
267
+ "hospital-department": 11,
268
+ "hospital-phone": 12,
269
+ "hospital-postcode": 13,
270
+ "hotel-address": 14,
271
+ "hotel-area": 15,
272
+ "hotel-book day": 16,
273
+ "hotel-book people": 17,
274
+ "hotel-book stay": 18,
275
+ "hotel-choice": 19,
276
+ "hotel-internet": 20,
277
+ "hotel-name": 21,
278
+ "hotel-parking": 22,
279
+ "hotel-phone": 23,
280
+ "hotel-postcode": 24,
281
+ "hotel-price range": 25,
282
+ "hotel-ref": 26,
283
+ "hotel-stars": 27,
284
+ "hotel-type": 28,
285
+ "police-address": 58,
286
+ "police-name": 59,
287
+ "police-phone": 60,
288
+ "police-postcode": 61,
289
+ "restaurant-address": 29,
290
+ "restaurant-area": 30,
291
+ "restaurant-book day": 31,
292
+ "restaurant-book people": 32,
293
+ "restaurant-book time": 33,
294
+ "restaurant-choice": 34,
295
+ "restaurant-food": 35,
296
+ "restaurant-name": 36,
297
+ "restaurant-phone": 37,
298
+ "restaurant-postcode": 38,
299
+ "restaurant-price range": 39,
300
+ "restaurant-ref": 40,
301
+ "taxi-arrive by": 41,
302
+ "taxi-departure": 42,
303
+ "taxi-destination": 43,
304
+ "taxi-leave at": 44,
305
+ "taxi-phone": 45,
306
+ "taxi-type": 46,
307
+ "train-arrive by": 47,
308
+ "train-book people": 48,
309
+ "train-choice": 49,
310
+ "train-day": 50,
311
+ "train-departure": 51,
312
+ "train-destination": 52,
313
+ "train-duration": 53,
314
+ "train-leave at": 54,
315
+ "train-price": 55,
316
+ "train-ref": 56,
317
+ "train-train id": 57
318
+ },
319
+ "test_batch_size": 16,
320
+ "torch_dtype": "float32",
321
+ "train_batch_size": 3,
322
+ "transformers_version": "4.24.0",
323
+ "type_vocab_size": 1,
324
+ "use_cache": true,
325
+ "use_descriptions": true,
326
+ "user_general_act_loss_weight": 0.2,
327
+ "user_goal_loss_weight": 1.0,
328
+ "user_request_loss_weight": 0.2,
329
+ "vocab_size": 50265
330
+ }