gokuls's picture
Training in progress, epoch 0
a41f12e 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": "alarm_remove",
"1": "music_dislikeness",
"10": "social_query",
"11": "play_radio",
"12": "wemo_off",
"13": "post",
"14": "transport_ticket",
"15": "cleaning",
"16": "audio_volume_mute",
"17": "general_greet",
"18": "podcasts",
"19": "cooking_query",
"2": "remove",
"20": "joke",
"21": "factoid",
"22": "calendar_set",
"23": "calendar_remove",
"24": "takeaway_order",
"25": "transport_traffic",
"26": "wemo_on",
"27": "currency",
"28": "iot_hue_lightup",
"29": "datetime_convert",
"3": "play_music",
"30": "alarm_set",
"31": "email_addcontact",
"32": "play_game",
"33": "recommendation_locations",
"34": "audio_volume_down",
"35": "hue_lightdim",
"36": "music",
"37": "qa_definition",
"38": "iot_hue_lightchange",
"39": "lists_createoradd",
"4": "traffic",
"40": "music_likeness",
"41": "datetime_query",
"42": "general_quirky",
"43": "news_query",
"44": "iot_hue_lighton",
"45": "qa_maths",
"46": "events",
"47": "iot_wemo_on",
"48": "query",
"49": "definition",
"5": "takeaway_query",
"50": "audio_volume_other",
"51": "ticket",
"52": "querycontact",
"53": "addcontact",
"54": "email_querycontact",
"55": "social_post",
"56": "recommendation_events",
"57": "iot_wemo_off",
"58": "locations",
"59": "play_podcasts",
"6": "alarm_query",
"60": "lists_query",
"61": "createoradd",
"62": "email_query",
"63": "transport_query",
"64": "coffee",
"65": "qa_stock",
"66": "music_settings",
"67": "play_audiobook",
"68": "iot_cleaning",
"69": "quirky",
"7": "iot_hue_lightdim",
"70": "volume_other",
"71": "greet",
"72": "weather_query",
"73": "music_query",
"74": "recommendation_movies",
"75": "general_joke",
"76": "transport_taxi",
"77": "radio",
"78": "hue_lightup",
"79": "calendar_query",
"8": "convert",
"80": "cooking_recipe",
"81": "game",
"82": "email_sendemail",
"83": "sendemail",
"84": "qa_factoid",
"85": "iot_coffee",
"86": "settings",
"87": "iot_hue_lightoff",
"88": "audio_volume_up",
"89": "hue_lightoff",
"9": "qa_currency",
"90": "lists_remove",
"91": "set"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"addcontact": "53",
"alarm_query": "6",
"alarm_remove": "0",
"alarm_set": "30",
"audio_volume_down": "34",
"audio_volume_mute": "16",
"audio_volume_other": "50",
"audio_volume_up": "88",
"calendar_query": "79",
"calendar_remove": "23",
"calendar_set": "22",
"cleaning": "15",
"coffee": "64",
"convert": "8",
"cooking_query": "19",
"cooking_recipe": "80",
"createoradd": "61",
"currency": "27",
"datetime_convert": "29",
"datetime_query": "41",
"definition": "49",
"email_addcontact": "31",
"email_query": "62",
"email_querycontact": "54",
"email_sendemail": "82",
"events": "46",
"factoid": "21",
"game": "81",
"general_greet": "17",
"general_joke": "75",
"general_quirky": "42",
"greet": "71",
"hue_lightdim": "35",
"hue_lightoff": "89",
"hue_lightup": "78",
"iot_cleaning": "68",
"iot_coffee": "85",
"iot_hue_lightchange": "38",
"iot_hue_lightdim": "7",
"iot_hue_lightoff": "87",
"iot_hue_lighton": "44",
"iot_hue_lightup": "28",
"iot_wemo_off": "57",
"iot_wemo_on": "47",
"joke": "20",
"lists_createoradd": "39",
"lists_query": "60",
"lists_remove": "90",
"locations": "58",
"music": "36",
"music_dislikeness": "1",
"music_likeness": "40",
"music_query": "73",
"music_settings": "66",
"news_query": "43",
"play_audiobook": "67",
"play_game": "32",
"play_music": "3",
"play_podcasts": "59",
"play_radio": "11",
"podcasts": "18",
"post": "13",
"qa_currency": "9",
"qa_definition": "37",
"qa_factoid": "84",
"qa_maths": "45",
"qa_stock": "65",
"query": "48",
"querycontact": "52",
"quirky": "69",
"radio": "77",
"recommendation_events": "56",
"recommendation_locations": "33",
"recommendation_movies": "74",
"remove": "2",
"sendemail": "83",
"set": "91",
"settings": "86",
"social_post": "55",
"social_query": "10",
"takeaway_order": "24",
"takeaway_query": "5",
"ticket": "51",
"traffic": "4",
"transport_query": "63",
"transport_taxi": "76",
"transport_ticket": "14",
"transport_traffic": "25",
"volume_other": "70",
"weather_query": "72",
"wemo_off": "12",
"wemo_on": "26"
},
"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.37.2",
"use_weighted_layer_sum": false,
"vocab_size": 32
}