LightChen2333 commited on
Commit
ac2c1ad
1 Parent(s): 4e2fca4

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.json +14 -52
  2. pytorch_model.bin +2 -2
  3. tokenizer.pkl +2 -2
config.json CHANGED
@@ -303,76 +303,38 @@
303
  ],
304
  "is_decoder": true,
305
  "model": {
306
- "_model_target_": "model.OpenSLUModel",
307
  "decoder": {
308
- "_model_target_": "model.decoder.StackPropagationDecoder",
309
  "intent_classifier": {
310
- "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
311
- "bidirectional": false,
312
- "dropout_rate": 0.5,
313
- "embedding_dim": 8,
314
- "force_ratio": 0.9,
315
- "hidden_dim": 64,
316
  "ignore_index": -100,
317
- "input_dim": 384,
318
  "intent_label_num": 21,
319
- "layer_num": 1,
320
- "mode": "token-level-intent",
321
- "return_sentence_level": true,
322
- "use_intent": true,
323
- "use_multi": false
324
- },
325
- "interaction": {
326
- "_model_target_": "model.decoder.interaction.StackInteraction",
327
- "differentiable": false,
328
- "input_dim": 384,
329
- "intent_label_num": 21,
330
- "slot_label_num": 120
331
  },
332
  "slot_classifier": {
333
- "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
334
- "bidirectional": false,
335
- "dropout_rate": 0.55,
336
- "embedding_dim": 32,
337
- "force_ratio": 0.9,
338
- "hidden_dim": 64,
339
  "ignore_index": -100,
340
- "input_dim": 405,
341
- "layer_num": 1,
342
  "mode": "slot",
343
- "return_sentence_level": false,
344
  "slot_label_num": 120,
345
- "use_multi": false,
346
  "use_slot": true
347
  }
348
  },
349
  "encoder": {
350
  "_model_target_": "model.encoder.AutoEncoder",
351
- "attention": {
352
- "dropout_rate": 0.6,
353
- "hidden_dim": 1024,
354
- "output_dim": 128
355
- },
356
- "embedding": {
357
- "dropout_rate": 0.55,
358
- "embedding_dim": 256,
359
- "vocab_size": 900
360
- },
361
- "encoder_name": "self-attention-lstm",
362
- "lstm": {
363
- "bidirectional": true,
364
- "dropout_rate": 0.5,
365
- "layer_num": 1,
366
- "output_dim": 256
367
- },
368
- "output_dim": 384,
369
- "return_sentence_level_hidden": false,
370
  "return_with_input": true
371
- }
 
372
  },
373
  "return_dict": false,
374
  "tokenizer_class": "OpenSLUv1",
375
  "torch_dtype": "float32",
376
  "transformers_version": "4.25.1",
377
- "vocab_size": 900
378
  }
 
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": 256,
313
  "intent_label_num": 21,
314
+ "mode": "intent",
315
+ "use_intent": true
 
 
 
 
 
 
 
 
 
 
316
  },
317
  "slot_classifier": {
318
+ "_model_target_": "model.decoder.classifier.LinearClassifier",
 
 
 
 
 
319
  "ignore_index": -100,
320
+ "input_dim": 256,
 
321
  "mode": "slot",
 
322
  "slot_label_num": 120,
 
323
  "use_slot": true
324
  }
325
  },
326
  "encoder": {
327
  "_model_target_": "model.encoder.AutoEncoder",
328
+ "encoder_name": "google/electra-small-discriminator",
329
+ "output_dim": 256,
330
+ "return_sentence_level_hidden": true,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
331
  "return_with_input": true
332
+ },
333
+ "ignore_index": -100
334
  },
335
  "return_dict": false,
336
  "tokenizer_class": "OpenSLUv1",
337
  "torch_dtype": "float32",
338
  "transformers_version": "4.25.1",
339
+ "vocab_size": 30522
340
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5d7caa64824e88f00d72f6dcb994550b6a8fec54cb4dafa345e069642255e8d7
3
- size 5786348
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77bb70493bfe029fe320502bfb7409f69aa5e6bfe3cd12d03bf9fbab44f7098f
3
+ size 54215971
tokenizer.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:91a3b819f2a4f8d82698577c606d5b7f832f808f48748d63c522e41a0802810c
3
- size 14048
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3aaf418fa585e9e145796b5150987aa208cc6e602eed0d0a563031e5c7918cb4
3
+ size 467165