mert_cmp_single / config.json
kurosekurose's picture
Upload config.json with huggingface_hub
bab1a9d verified
{
"architectures": [
"MertForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": null,
"conv_kernel": [
10,
3,
3,
3,
3,
2,
2
],
"conv_stride": [
5,
2,
2,
2,
2,
2,
2
],
"feature_extractor_type": "Wav2Vec2FeatureExtractor",
"feat_extract_activation": "gelu",
"feat_extract_norm": "group",
"feat_proj_dropout": 0.0,
"final_dropout": 0.1,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"initializer_range": 0.02,
"intermediate_size": 3072,
"layer_norm_eps": 1e-05,
"layerdrop": 0.1,
"mask_feature_length": 10,
"mask_feature_prob": 0.0,
"mask_time_length": 10,
"mask_time_prob": 0.05,
"model_type": "mert",
"num_attention_heads": 12,
"num_conv_pos_embeddings": 128,
"num_conv_pos_embedding_groups": 16,
"num_feat_extract_layers": 7,
"num_hidden_layers": 12,
"num_labels": 9,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"transformers_version": "4.30.0",
"use_weighted_layer_sum": false,
"vocab_size": 32,
"id2label": {
"0": "ζ₯½γ—い",
"1": "η››γ‚ŠδΈŠγŒγ‚‹",
"2": "怖い",
"3": "ドキドキする",
"4": "ハラハラする",
"5": "悲しい",
"6": "εˆ‡γͺい",
"7": "γƒͺラックスする",
"8": "η™’γ•γ‚Œγ‚‹"
},
"label2id": {
"ζ₯½γ—い": 0,
"η››γ‚ŠδΈŠγŒγ‚‹": 1,
"怖い": 2,
"ドキドキする": 3,
"ハラハラする": 4,
"悲しい": 5,
"εˆ‡γͺい": 6,
"γƒͺラックスする": 7,
"η™’γ•γ‚Œγ‚‹": 8
}
}