gokuls's picture
Training in progress, epoch 0
42edfc9 verified
raw
history blame
6.32 kB
{
"_name_or_path": "facebook/hubert-base-ls960",
"activation_dropout": 0.1,
"apply_spec_augment": true,
"architectures": [
"HubertForSequenceClassification"
],
"attention_dropout": 0.1,
"bos_token_id": 1,
"classifier_proj_size": 256,
"conv_bias": false,
"conv_dim": [
512,
512,
512,
512,
512,
512,
512
],
"conv_kernel": [
10,
3,
3,
3,
3,
2,
2
],
"conv_stride": [
5,
2,
2,
2,
2,
2,
2
],
"ctc_loss_reduction": "sum",
"ctc_zero_infinity": false,
"do_stable_layer_norm": false,
"eos_token_id": 2,
"feat_extract_activation": "gelu",
"feat_extract_dropout": 0.0,
"feat_extract_norm": "group",
"feat_proj_dropout": 0.1,
"feat_proj_layer_norm": true,
"final_dropout": 0.1,
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout": 0.1,
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"id2label": {
"0": "set",
"1": "takeaway_query",
"10": "wemo_off",
"11": "recommendation_movies",
"12": "datetime_convert",
"13": "cooking_query",
"14": "play_audiobook",
"15": "quirky",
"16": "joke",
"17": "datetime_query",
"18": "remove",
"19": "transport_taxi",
"2": "greet",
"20": "lists_query",
"21": "email_query",
"22": "recommendation_locations",
"23": "social_post",
"24": "convert",
"25": "sendemail",
"26": "calendar_query",
"27": "audio_volume_down",
"28": "social_query",
"29": "qa_currency",
"3": "ticket",
"30": "lists_remove",
"31": "music_likeness",
"32": "iot_coffee",
"33": "music_settings",
"34": "email_sendemail",
"35": "iot_hue_lightoff",
"36": "email_querycontact",
"37": "locations",
"38": "radio",
"39": "settings",
"4": "post",
"40": "qa_stock",
"41": "iot_wemo_off",
"42": "email_addcontact",
"43": "audio_volume_mute",
"44": "hue_lightdim",
"45": "events",
"46": "currency",
"47": "calendar_set",
"48": "general_greet",
"49": "factoid",
"5": "createoradd",
"50": "hue_lightup",
"51": "querycontact",
"52": "transport_traffic",
"53": "general_quirky",
"54": "hue_lightoff",
"55": "recommendation_events",
"56": "transport_ticket",
"57": "iot_hue_lighton",
"58": "music",
"59": "iot_hue_lightchange",
"6": "takeaway_order",
"60": "alarm_query",
"61": "weather_query",
"62": "definition",
"63": "iot_wemo_on",
"64": "coffee",
"65": "qa_factoid",
"66": "podcasts",
"67": "qa_maths",
"68": "calendar_remove",
"69": "cooking_recipe",
"7": "transport_query",
"70": "query",
"71": "audio_volume_other",
"72": "addcontact",
"73": "alarm_set",
"74": "cleaning",
"75": "lists_createoradd",
"76": "volume_other",
"77": "alarm_remove",
"78": "play_radio",
"79": "traffic",
"8": "qa_definition",
"80": "play_music",
"81": "game",
"82": "wemo_on",
"83": "music_query",
"84": "general_joke",
"85": "news_query",
"86": "iot_hue_lightup",
"87": "audio_volume_up",
"88": "play_game",
"89": "music_dislikeness",
"9": "play_podcasts",
"90": "iot_hue_lightdim",
"91": "iot_cleaning"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"addcontact": "72",
"alarm_query": "60",
"alarm_remove": "77",
"alarm_set": "73",
"audio_volume_down": "27",
"audio_volume_mute": "43",
"audio_volume_other": "71",
"audio_volume_up": "87",
"calendar_query": "26",
"calendar_remove": "68",
"calendar_set": "47",
"cleaning": "74",
"coffee": "64",
"convert": "24",
"cooking_query": "13",
"cooking_recipe": "69",
"createoradd": "5",
"currency": "46",
"datetime_convert": "12",
"datetime_query": "17",
"definition": "62",
"email_addcontact": "42",
"email_query": "21",
"email_querycontact": "36",
"email_sendemail": "34",
"events": "45",
"factoid": "49",
"game": "81",
"general_greet": "48",
"general_joke": "84",
"general_quirky": "53",
"greet": "2",
"hue_lightdim": "44",
"hue_lightoff": "54",
"hue_lightup": "50",
"iot_cleaning": "91",
"iot_coffee": "32",
"iot_hue_lightchange": "59",
"iot_hue_lightdim": "90",
"iot_hue_lightoff": "35",
"iot_hue_lighton": "57",
"iot_hue_lightup": "86",
"iot_wemo_off": "41",
"iot_wemo_on": "63",
"joke": "16",
"lists_createoradd": "75",
"lists_query": "20",
"lists_remove": "30",
"locations": "37",
"music": "58",
"music_dislikeness": "89",
"music_likeness": "31",
"music_query": "83",
"music_settings": "33",
"news_query": "85",
"play_audiobook": "14",
"play_game": "88",
"play_music": "80",
"play_podcasts": "9",
"play_radio": "78",
"podcasts": "66",
"post": "4",
"qa_currency": "29",
"qa_definition": "8",
"qa_factoid": "65",
"qa_maths": "67",
"qa_stock": "40",
"query": "70",
"querycontact": "51",
"quirky": "15",
"radio": "38",
"recommendation_events": "55",
"recommendation_locations": "22",
"recommendation_movies": "11",
"remove": "18",
"sendemail": "25",
"set": "0",
"settings": "39",
"social_post": "23",
"social_query": "28",
"takeaway_order": "6",
"takeaway_query": "1",
"ticket": "3",
"traffic": "79",
"transport_query": "7",
"transport_taxi": "19",
"transport_ticket": "56",
"transport_traffic": "52",
"volume_other": "76",
"weather_query": "61",
"wemo_off": "10",
"wemo_on": "82"
},
"layer_norm_eps": 1e-05,
"layerdrop": 0.1,
"mask_feature_length": 10,
"mask_feature_min_masks": 0,
"mask_feature_prob": 0.0,
"mask_time_length": 10,
"mask_time_min_masks": 2,
"mask_time_prob": 0.05,
"model_type": "hubert",
"num_attention_heads": 12,
"num_conv_pos_embedding_groups": 16,
"num_conv_pos_embeddings": 128,
"num_feat_extract_layers": 7,
"num_hidden_layers": 12,
"pad_token_id": 0,
"tokenizer_class": "Wav2Vec2CTCTokenizer",
"torch_dtype": "float32",
"transformers_version": "4.36.2",
"use_weighted_layer_sum": false,
"vocab_size": 32
}