{ "_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 }