LightChen2333 commited on
Commit
96016f8
1 Parent(s): e02fd23

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +29 -1
config.json CHANGED
@@ -299,9 +299,37 @@
299
  "slot": 120
300
  },
301
  "architectures": [
302
- "PretrainedModelForSLU"
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",