LightChen2333
commited on
Commit
•
96016f8
1
Parent(s):
e02fd23
Upload config.json
Browse files- config.json +29 -1
config.json
CHANGED
@@ -299,9 +299,37 @@
|
|
299 |
"slot": 120
|
300 |
},
|
301 |
"architectures": [
|
302 |
-
"
|
303 |
],
|
304 |
"is_decoder": true,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
305 |
"return_dict": false,
|
306 |
"tokenizer_class": "OpenSLUv1",
|
307 |
"torch_dtype": "float32",
|
|
|
299 |
"slot": 120
|
300 |
},
|
301 |
"architectures": [
|
302 |
+
"PretrainedModelForSLUToSave"
|
303 |
],
|
304 |
"is_decoder": true,
|
305 |
+
"model": {
|
306 |
+
"_model_target_": "model.open_slu_model.OpenSLUModel",
|
307 |
+
"decoder": {
|
308 |
+
"_model_target_": "model.decoder.base_decoder.BaseDecoder",
|
309 |
+
"intent_classifier": {
|
310 |
+
"_model_target_": "model.decoder.classifier.LinearClassifier",
|
311 |
+
"ignore_index": -100,
|
312 |
+
"input_dim": 768,
|
313 |
+
"mode": "intent",
|
314 |
+
"use_intent": true
|
315 |
+
},
|
316 |
+
"slot_classifier": {
|
317 |
+
"_model_target_": "model.decoder.classifier.LinearClassifier",
|
318 |
+
"ignore_index": -100,
|
319 |
+
"input_dim": 768,
|
320 |
+
"mode": "slot",
|
321 |
+
"use_slot": true
|
322 |
+
}
|
323 |
+
},
|
324 |
+
"encoder": {
|
325 |
+
"_model_target_": "model.encoder.AutoEncoder",
|
326 |
+
"encoder_name": "bert-base-uncased",
|
327 |
+
"output_dim": 768,
|
328 |
+
"return_sentence_level_hidden": true,
|
329 |
+
"return_with_input": true
|
330 |
+
},
|
331 |
+
"ignore_index": -100
|
332 |
+
},
|
333 |
"return_dict": false,
|
334 |
"tokenizer_class": "OpenSLUv1",
|
335 |
"torch_dtype": "float32",
|