| { | |
| "_name_or_path": "ConvLab/setsumbt-dst_nlu-multiwoz21-EnD2", | |
| "active_domain_loss_weight": 0.2, | |
| "architectures": [ | |
| "RobertaSetSUMBT" | |
| ], | |
| "attention_probs_dropout_prob": 0.1, | |
| "bos_token_id": 0, | |
| "candidate_embedding_model_name": "roberta-base", | |
| "classifier_dropout": null, | |
| "code_version": "setsumbt/a4dd9b74f22d0ef72c9fe7bcf125eb61d3f5f4f7", | |
| "dev_batch_size": 12, | |
| "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": [ | |
| 58, | |
| 59, | |
| 60, | |
| 61 | |
| ], | |
| "restaurant": [ | |
| 32, | |
| 33, | |
| 34, | |
| 35, | |
| 36, | |
| 37, | |
| 38, | |
| 39, | |
| 40, | |
| 29, | |
| 30, | |
| 31 | |
| ], | |
| "taxi": [ | |
| 41, | |
| 42, | |
| 43, | |
| 44, | |
| 45, | |
| 46 | |
| ], | |
| "train": [ | |
| 47, | |
| 48, | |
| 49, | |
| 50, | |
| 51, | |
| 52, | |
| 53, | |
| 54, | |
| 55, | |
| 56, | |
| 57 | |
| ] | |
| }, | |
| "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": 30, | |
| "restaurant-book day": 31, | |
| "restaurant-book people": 32, | |
| "restaurant-book time": 33, | |
| "restaurant-food": 35, | |
| "restaurant-name": 36, | |
| "restaurant-price range": 39, | |
| "taxi-arrive by": 41, | |
| "taxi-departure": 42, | |
| "taxi-destination": 43, | |
| "taxi-leave at": 44, | |
| "train-arrive by": 47, | |
| "train-book people": 48, | |
| "train-day": 50, | |
| "train-departure": 51, | |
| "train-destination": 52, | |
| "train-leave at": 54 | |
| }, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 3072, | |
| "layer_norm_eps": 1e-05, | |
| "loss_function": "distribution_distillation", | |
| "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": true, | |
| "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": 58, | |
| "police-phone": 60, | |
| "police-postcode": 61, | |
| "restaurant-address": 29, | |
| "restaurant-area": 30, | |
| "restaurant-food": 35, | |
| "restaurant-phone": 37, | |
| "restaurant-postcode": 38, | |
| "restaurant-price range": 39, | |
| "restaurant-ref": 40, | |
| "taxi-phone": 45, | |
| "taxi-type": 46, | |
| "train-arrive by": 47, | |
| "train-duration": 53, | |
| "train-leave at": 54, | |
| "train-price": 55, | |
| "train-ref": 56, | |
| "train-train id": 57 | |
| }, | |
| "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": 58, | |
| "police-name": 59, | |
| "police-phone": 60, | |
| "police-postcode": 61, | |
| "restaurant-address": 29, | |
| "restaurant-area": 30, | |
| "restaurant-book day": 31, | |
| "restaurant-book people": 32, | |
| "restaurant-book time": 33, | |
| "restaurant-choice": 34, | |
| "restaurant-food": 35, | |
| "restaurant-name": 36, | |
| "restaurant-phone": 37, | |
| "restaurant-postcode": 38, | |
| "restaurant-price range": 39, | |
| "restaurant-ref": 40, | |
| "taxi-arrive by": 41, | |
| "taxi-departure": 42, | |
| "taxi-destination": 43, | |
| "taxi-leave at": 44, | |
| "taxi-phone": 45, | |
| "taxi-type": 46, | |
| "train-arrive by": 47, | |
| "train-book people": 48, | |
| "train-choice": 49, | |
| "train-day": 50, | |
| "train-departure": 51, | |
| "train-destination": 52, | |
| "train-duration": 53, | |
| "train-leave at": 54, | |
| "train-price": 55, | |
| "train-ref": 56, | |
| "train-train id": 57 | |
| }, | |
| "test_batch_size": 16, | |
| "torch_dtype": "float32", | |
| "train_batch_size": 3, | |
| "transformers_version": "4.24.0", | |
| "type_vocab_size": 1, | |
| "use_cache": true, | |
| "use_descriptions": true, | |
| "user_general_act_loss_weight": 0.2, | |
| "user_goal_loss_weight": 1.0, | |
| "user_request_loss_weight": 0.2, | |
| "vocab_size": 50265 | |
| } | |