| { |
| "_name_or_path": "ConvLab/setsumbt-dst-multiwoz21", |
| "adapters": { |
| "adapters": {}, |
| "config_map": {}, |
| "fusion_config_map": {}, |
| "fusions": {} |
| }, |
| "architectures": [ |
| "RobertaSetSUMBT" |
| ], |
| "attention_probs_dropout_prob": 0.1, |
| "bos_token_id": 0, |
| "candidate_embedding_model_name": "roberta-base", |
| "classifier_dropout": null, |
| "dev_batch_size": 16, |
| "distance_measure": "cosine", |
| "domain_ids": { |
| "attraction": [ |
| 1, |
| 2, |
| 3, |
| 4, |
| 5, |
| 6, |
| 7, |
| 8, |
| 9 |
| ], |
| "hospital": [ |
| 10, |
| 11, |
| 12, |
| 13 |
| ], |
| "hotel": [ |
| 14, |
| 15, |
| 16, |
| 17, |
| 18, |
| 19, |
| 20, |
| 21, |
| 22, |
| 23, |
| 24, |
| 25, |
| 26, |
| 27, |
| 28 |
| ], |
| "police": [ |
| 32, |
| 29, |
| 30, |
| 31 |
| ], |
| "restaurant": [ |
| 33, |
| 34, |
| 35, |
| 36, |
| 37, |
| 38, |
| 39, |
| 40, |
| 41, |
| 42, |
| 43, |
| 44 |
| ], |
| "taxi": [ |
| 45, |
| 46, |
| 47, |
| 48, |
| 49, |
| 50 |
| ], |
| "train": [ |
| 51, |
| 52, |
| 53, |
| 54, |
| 55, |
| 56, |
| 57, |
| 58, |
| 59, |
| 60, |
| 61 |
| ] |
| }, |
| "dropout_rate": 0.3, |
| "eos_token_id": 2, |
| "freeze_encoder": false, |
| "hidden_act": "gelu", |
| "hidden_dropout_prob": 0.1, |
| "hidden_size": 768, |
| "informable_slot_ids": { |
| "attraction-area": 2, |
| "attraction-name": 5, |
| "attraction-type": 9, |
| "hospital-department": 11, |
| "hotel-area": 15, |
| "hotel-book day": 16, |
| "hotel-book people": 17, |
| "hotel-book stay": 18, |
| "hotel-internet": 20, |
| "hotel-name": 21, |
| "hotel-parking": 22, |
| "hotel-price range": 25, |
| "hotel-stars": 27, |
| "hotel-type": 28, |
| "restaurant-area": 34, |
| "restaurant-book day": 35, |
| "restaurant-book people": 36, |
| "restaurant-book time": 37, |
| "restaurant-food": 39, |
| "restaurant-name": 40, |
| "restaurant-price range": 43, |
| "taxi-arrive by": 45, |
| "taxi-departure": 46, |
| "taxi-destination": 47, |
| "taxi-leave at": 48, |
| "train-arrive by": 51, |
| "train-book people": 52, |
| "train-day": 54, |
| "train-departure": 55, |
| "train-destination": 56, |
| "train-leave at": 58 |
| }, |
| "initializer_range": 0.02, |
| "intermediate_size": 3072, |
| "label_smoothing": 0.05, |
| "layer_norm_eps": 1e-05, |
| "loss_function": "labelsmoothing", |
| "max_candidate_len": 12, |
| "max_dialogue_len": 12, |
| "max_position_embeddings": 514, |
| "max_slot_len": 12, |
| "max_turn_len": 64, |
| "model_type": "roberta", |
| "nbt_hidden_size": 300, |
| "nbt_layers": 1, |
| "nbt_type": "gru", |
| "num_attention_heads": 12, |
| "num_hidden_layers": 12, |
| "num_slots": 62, |
| "num_values": { |
| "attraction-address": 1, |
| "attraction-area": 7, |
| "attraction-choice": 1, |
| "attraction-entrance fee": 1, |
| "attraction-name": 156, |
| "attraction-open hours": 1, |
| "attraction-phone": 1, |
| "attraction-postcode": 1, |
| "attraction-type": 14, |
| "hospital-address": 1, |
| "hospital-department": 49, |
| "hospital-phone": 1, |
| "hospital-postcode": 1, |
| "hotel-address": 1, |
| "hotel-area": 7, |
| "hotel-book day": 9, |
| "hotel-book people": 12, |
| "hotel-book stay": 12, |
| "hotel-choice": 1, |
| "hotel-internet": 5, |
| "hotel-name": 72, |
| "hotel-parking": 5, |
| "hotel-phone": 1, |
| "hotel-postcode": 1, |
| "hotel-price range": 5, |
| "hotel-ref": 1, |
| "hotel-stars": 8, |
| "hotel-type": 5, |
| "police-address": 1, |
| "police-name": 1, |
| "police-phone": 1, |
| "police-postcode": 1, |
| "restaurant-address": 1, |
| "restaurant-area": 7, |
| "restaurant-book day": 9, |
| "restaurant-book people": 12, |
| "restaurant-book time": 290, |
| "restaurant-choice": 1, |
| "restaurant-food": 103, |
| "restaurant-name": 185, |
| "restaurant-phone": 1, |
| "restaurant-postcode": 1, |
| "restaurant-price range": 5, |
| "restaurant-ref": 1, |
| "taxi-arrive by": 290, |
| "taxi-departure": 255, |
| "taxi-destination": 253, |
| "taxi-leave at": 290, |
| "taxi-phone": 1, |
| "taxi-type": 1, |
| "train-arrive by": 290, |
| "train-book people": 12, |
| "train-choice": 1, |
| "train-day": 9, |
| "train-departure": 32, |
| "train-destination": 28, |
| "train-duration": 1, |
| "train-leave at": 290, |
| "train-price": 1, |
| "train-ref": 1, |
| "train-train id": 1 |
| }, |
| "pad_token_id": 1, |
| "position_embedding_type": "absolute", |
| "predict_actions": false, |
| "requestable_slot_ids": { |
| "attraction-address": 1, |
| "attraction-area": 2, |
| "attraction-entrance fee": 4, |
| "attraction-phone": 7, |
| "attraction-postcode": 8, |
| "attraction-type": 9, |
| "hospital-address": 10, |
| "hospital-phone": 12, |
| "hospital-postcode": 13, |
| "hotel-address": 14, |
| "hotel-area": 15, |
| "hotel-internet": 20, |
| "hotel-parking": 22, |
| "hotel-phone": 23, |
| "hotel-postcode": 24, |
| "hotel-price range": 25, |
| "hotel-ref": 26, |
| "hotel-stars": 27, |
| "hotel-type": 28, |
| "police-address": 29, |
| "police-phone": 31, |
| "police-postcode": 32, |
| "restaurant-address": 33, |
| "restaurant-area": 34, |
| "restaurant-food": 39, |
| "restaurant-phone": 41, |
| "restaurant-postcode": 42, |
| "restaurant-price range": 43, |
| "restaurant-ref": 44, |
| "taxi-phone": 49, |
| "taxi-type": 50, |
| "train-arrive by": 51, |
| "train-duration": 57, |
| "train-leave at": 58, |
| "train-price": 59, |
| "train-ref": 60, |
| "train-train id": 61 |
| }, |
| "rnn_zero_init": false, |
| "seed": 0, |
| "set_pooling": "cnn", |
| "set_similarity": true, |
| "slot_attention_heads": 12, |
| "slot_ids": { |
| "attraction-address": 1, |
| "attraction-area": 2, |
| "attraction-choice": 3, |
| "attraction-entrance fee": 4, |
| "attraction-name": 5, |
| "attraction-open hours": 6, |
| "attraction-phone": 7, |
| "attraction-postcode": 8, |
| "attraction-type": 9, |
| "hospital-address": 10, |
| "hospital-department": 11, |
| "hospital-phone": 12, |
| "hospital-postcode": 13, |
| "hotel-address": 14, |
| "hotel-area": 15, |
| "hotel-book day": 16, |
| "hotel-book people": 17, |
| "hotel-book stay": 18, |
| "hotel-choice": 19, |
| "hotel-internet": 20, |
| "hotel-name": 21, |
| "hotel-parking": 22, |
| "hotel-phone": 23, |
| "hotel-postcode": 24, |
| "hotel-price range": 25, |
| "hotel-ref": 26, |
| "hotel-stars": 27, |
| "hotel-type": 28, |
| "police-address": 29, |
| "police-name": 30, |
| "police-phone": 31, |
| "police-postcode": 32, |
| "restaurant-address": 33, |
| "restaurant-area": 34, |
| "restaurant-book day": 35, |
| "restaurant-book people": 36, |
| "restaurant-book time": 37, |
| "restaurant-choice": 38, |
| "restaurant-food": 39, |
| "restaurant-name": 40, |
| "restaurant-phone": 41, |
| "restaurant-postcode": 42, |
| "restaurant-price range": 43, |
| "restaurant-ref": 44, |
| "taxi-arrive by": 45, |
| "taxi-departure": 46, |
| "taxi-destination": 47, |
| "taxi-leave at": 48, |
| "taxi-phone": 49, |
| "taxi-type": 50, |
| "train-arrive by": 51, |
| "train-book people": 52, |
| "train-choice": 53, |
| "train-day": 54, |
| "train-departure": 55, |
| "train-destination": 56, |
| "train-duration": 57, |
| "train-leave at": 58, |
| "train-price": 59, |
| "train-ref": 60, |
| "train-train id": 61 |
| }, |
| "test_batch_size": 16, |
| "tokenizer_name": "roberta-base", |
| "torch_dtype": "float32", |
| "train_batch_size": 8, |
| "transformers_version": "4.24.0", |
| "type_vocab_size": 1, |
| "use_cache": true, |
| "use_descriptions": true, |
| "vocab_size": 50265 |
| } |
|
|
