gokuls's picture
Training in progress, epoch 0
73ba9c0 verified
raw
history blame
7.12 kB
{
"_name_or_path": "facebook/wav2vec2-base",
"activation_dropout": 0.0,
"adapter_attn_dim": null,
"adapter_kernel_size": 3,
"adapter_stride": 2,
"add_adapter": false,
"apply_spec_augment": true,
"architectures": [
"Wav2Vec2ForSequenceClassification"
],
"attention_dropout": 0.1,
"bos_token_id": 1,
"classifier_proj_size": 256,
"codevector_dim": 256,
"contrastive_logits_temperature": 0.1,
"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,
"diversity_loss_weight": 0.1,
"do_stable_layer_norm": false,
"eos_token_id": 2,
"feat_extract_activation": "gelu",
"feat_extract_norm": "group",
"feat_proj_dropout": 0.1,
"feat_quantizer_dropout": 0.0,
"final_dropout": 0.0,
"freeze_feat_extract_train": true,
"hidden_act": "gelu",
"hidden_dropout": 0.1,
"hidden_size": 768,
"id2label": {
"0": "play_audiobook",
"1": "general_joke",
"10": "email_querycontact",
"11": "iot_hue_lightdim",
"12": "cleaning",
"13": "ticket",
"14": "alarm_set",
"15": "calendar_remove",
"16": "takeaway_order",
"17": "iot_cleaning",
"18": "transport_taxi",
"19": "recommendation_movies",
"2": "query",
"20": "querycontact",
"21": "iot_hue_lightoff",
"22": "wemo_off",
"23": "addcontact",
"24": "transport_ticket",
"25": "audio_volume_down",
"26": "hue_lightoff",
"27": "qa_definition",
"28": "podcasts",
"29": "play_radio",
"3": "factoid",
"30": "iot_hue_lighton",
"31": "iot_wemo_off",
"32": "wemo_on",
"33": "iot_coffee",
"34": "audio_volume_other",
"35": "lists_remove",
"36": "calendar_set",
"37": "social_query",
"38": "sendemail",
"39": "music_query",
"4": "iot_wemo_on",
"40": "events",
"41": "audio_volume_mute",
"42": "cooking_recipe",
"43": "radio",
"44": "qa_currency",
"45": "play_podcasts",
"46": "social_post",
"47": "hue_lightdim",
"48": "recommendation_events",
"49": "music_settings",
"5": "transport_traffic",
"50": "traffic",
"51": "cooking_query",
"52": "play_game",
"53": "qa_maths",
"54": "post",
"55": "play_music",
"56": "iot_hue_lightchange",
"57": "lists_createoradd",
"58": "alarm_query",
"59": "qa_factoid",
"6": "music_likeness",
"60": "joke",
"61": "convert",
"62": "news_query",
"63": "weather_query",
"64": "lists_query",
"65": "set",
"66": "game",
"67": "email_addcontact",
"68": "settings",
"69": "general_greet",
"7": "createoradd",
"70": "hue_lightup",
"71": "takeaway_query",
"72": "datetime_convert",
"73": "greet",
"74": "general_quirky",
"75": "quirky",
"76": "iot_hue_lightup",
"77": "calendar_query",
"78": "locations",
"79": "email_sendemail",
"8": "email_query",
"80": "definition",
"81": "audio_volume_up",
"82": "music",
"83": "datetime_query",
"84": "qa_stock",
"85": "coffee",
"86": "currency",
"87": "alarm_remove",
"88": "transport_query",
"89": "music_dislikeness",
"9": "remove",
"90": "volume_other",
"91": "recommendation_locations"
},
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"addcontact": "23",
"alarm_query": "58",
"alarm_remove": "87",
"alarm_set": "14",
"audio_volume_down": "25",
"audio_volume_mute": "41",
"audio_volume_other": "34",
"audio_volume_up": "81",
"calendar_query": "77",
"calendar_remove": "15",
"calendar_set": "36",
"cleaning": "12",
"coffee": "85",
"convert": "61",
"cooking_query": "51",
"cooking_recipe": "42",
"createoradd": "7",
"currency": "86",
"datetime_convert": "72",
"datetime_query": "83",
"definition": "80",
"email_addcontact": "67",
"email_query": "8",
"email_querycontact": "10",
"email_sendemail": "79",
"events": "40",
"factoid": "3",
"game": "66",
"general_greet": "69",
"general_joke": "1",
"general_quirky": "74",
"greet": "73",
"hue_lightdim": "47",
"hue_lightoff": "26",
"hue_lightup": "70",
"iot_cleaning": "17",
"iot_coffee": "33",
"iot_hue_lightchange": "56",
"iot_hue_lightdim": "11",
"iot_hue_lightoff": "21",
"iot_hue_lighton": "30",
"iot_hue_lightup": "76",
"iot_wemo_off": "31",
"iot_wemo_on": "4",
"joke": "60",
"lists_createoradd": "57",
"lists_query": "64",
"lists_remove": "35",
"locations": "78",
"music": "82",
"music_dislikeness": "89",
"music_likeness": "6",
"music_query": "39",
"music_settings": "49",
"news_query": "62",
"play_audiobook": "0",
"play_game": "52",
"play_music": "55",
"play_podcasts": "45",
"play_radio": "29",
"podcasts": "28",
"post": "54",
"qa_currency": "44",
"qa_definition": "27",
"qa_factoid": "59",
"qa_maths": "53",
"qa_stock": "84",
"query": "2",
"querycontact": "20",
"quirky": "75",
"radio": "43",
"recommendation_events": "48",
"recommendation_locations": "91",
"recommendation_movies": "19",
"remove": "9",
"sendemail": "38",
"set": "65",
"settings": "68",
"social_post": "46",
"social_query": "37",
"takeaway_order": "16",
"takeaway_query": "71",
"ticket": "13",
"traffic": "50",
"transport_query": "88",
"transport_taxi": "18",
"transport_ticket": "24",
"transport_traffic": "5",
"volume_other": "90",
"weather_query": "63",
"wemo_off": "22",
"wemo_on": "32"
},
"layer_norm_eps": 1e-05,
"layerdrop": 0.0,
"mask_channel_length": 10,
"mask_channel_min_space": 1,
"mask_channel_other": 0.0,
"mask_channel_prob": 0.0,
"mask_channel_selection": "static",
"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_min_space": 1,
"mask_time_other": 0.0,
"mask_time_prob": 0.05,
"mask_time_selection": "static",
"model_type": "wav2vec2",
"no_mask_channel_overlap": false,
"no_mask_time_overlap": false,
"num_adapter_layers": 3,
"num_attention_heads": 12,
"num_codevector_groups": 2,
"num_codevectors_per_group": 320,
"num_conv_pos_embedding_groups": 16,
"num_conv_pos_embeddings": 128,
"num_feat_extract_layers": 7,
"num_hidden_layers": 12,
"num_negatives": 100,
"output_hidden_size": 768,
"pad_token_id": 0,
"proj_codevector_dim": 256,
"tdnn_dilation": [
1,
2,
3,
1,
1
],
"tdnn_dim": [
512,
512,
512,
512,
1500
],
"tdnn_kernel": [
5,
3,
3,
1,
1
],
"torch_dtype": "float32",
"transformers_version": "4.37.2",
"use_weighted_layer_sum": false,
"vocab_size": 32,
"xvector_output_dim": 512
}