From edc42333c6612c7db3889c22410fac7de68f85d0 Mon Sep 17 00:00:00 2001 From: harshit katyal Date: Sun, 12 Dec 2021 20:22:48 +0000 Subject: [PATCH] Upload config.json --- config.json | 104 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 104 insertions(+) create mode 100644 config.json diff --git a/config.json b/config.json new file mode 100644 index 0000000..5aa897b --- /dev/null +++ b/config.json @@ -0,0 +1,104 @@ +{ + "_name_or_path": "lighteternal/wav2vec2-large-xlsr-53-greek", + "activation_dropout": 0.0, + "apply_spec_augment": true, + "architectures": [ + "Wav2Vec2ForSpeechClassification" + ], + "attention_dropout": 0.1, + "bos_token_id": 1, + "classifier_proj_size": 256, + "codevector_dim": 256, + "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": true, + "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": "wav2vec2_clf", + "gradient_checkpointing": true, + "hidden_act": "gelu", + "hidden_dropout": 0.1, + "hidden_size": 1024, + "id2label": { + "0": "anger", + "1": "disgust", + "2": "fear", + "3": "happiness", + "4": "sadness" + }, + "initializer_range": 0.02, + "intermediate_size": 4096, + "label2id": { + "anger": 0, + "disgust": 1, + "fear": 2, + "happiness": 3, + "sadness": 4 + }, + "layer_norm_eps": 1e-05, + "layerdrop": 0.1, + "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_prob": 0.0, + "mask_time_length": 10, + "mask_time_min_space": 1, + "mask_time_other": 0.0, + "mask_time_prob": 0.05, + "mask_time_selection": "static", + "model_type": "wav2vec2", + "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": 24, + "num_negatives": 100, + "pad_token_id": 54, + "pooling_mode": "mean", + "problem_type": "single_label_classification", + "proj_codevector_dim": 256, + "torch_dtype": "float32", + "transformers_version": "4.11.0.dev0", + "use_weighted_layer_sum": false, + "vocab_size": 55 +}