{ "_name_or_path": "/fsx/yoach/accent_classification_output_final_v3_smaller_dataset/", "activation_dropout": 0.0, "adapter_attn_dim": 16, "adapter_kernel_size": 3, "adapter_stride": 2, "add_adapter": false, "apply_spec_augment": true, "architectures": [ "Wav2Vec2ForSequenceClassification" ], "attention_dropout": 0.0, "bos_token_id": 1, "classifier_proj_size": 1024, "codevector_dim": 1024, "contrastive_logits_temperature": 0.1, "conv_bias": true, "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": "mean", "ctc_zero_infinity": false, "diversity_loss_weight": 0.1, "do_stable_layer_norm": true, "eos_token_id": 2, "feat_extract_activation": "gelu", "feat_extract_dropout": 0.0, "feat_extract_norm": "layer", "feat_proj_dropout": 0.0, "feat_quantizer_dropout": 0.0, "final_dropout": 0.0, "finetuning_task": "audio-classification", "hidden_act": "gelu", "hidden_dropout": 0.0, "hidden_size": 1280, "id2label": { "0": "American", "1": "Australian", "2": "Canadian", "3": "Chinese", "4": "Czech", "5": "Dutch", "6": "Eastern european", "7": "English", "8": "Estonian", "9": "Finnish", "10": "French", "11": "German", "12": "Hungarian", "13": "Indian", "14": "Irish", "15": "Italian", "16": "Jamaican", "17": "Latin american", "18": "Malaysian", "19": "New zealand", "20": "Polish", "21": "Romanian", "22": "Scottish", "23": "Singaporean", "24": "Slovak", "25": "South african", "26": "Spanish", "27": "Welsh" }, "initializer_range": 0.02, "intermediate_size": 5120, "label2id": { "American": "0", "Australian": "1", "Canadian": "2", "Chinese": "3", "Czech": "4", "Dutch": "5", "Eastern european": "6", "English": "7", "Estonian": "8", "Finnish": "9", "French": "10", "German": "11", "Hungarian": "12", "Indian": "13", "Irish": "14", "Italian": "15", "Jamaican": "16", "Latin american": "17", "Malaysian": "18", "New zealand": "19", "Polish": "20", "Romanian": "21", "Scottish": "22", "Singaporean": "23", "Slovak": "24", "South african": "25", "Spanish": "26", "Welsh": "27" }, "layer_norm_eps": 1e-05, "layerdrop": 0.0, "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": "wav2vec2", "num_adapter_layers": 3, "num_attention_heads": 16, "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": 48, "num_negatives": 100, "output_hidden_size": 1280, "pad_token_id": 0, "proj_codevector_dim": 1024, "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.40.2", "use_weighted_layer_sum": false, "vocab_size": 154, "xvector_output_dim": 512 }