From 2838dc51b12ea6e9fb854049d46137561ebf0c03 Mon Sep 17 00:00:00 2001 From: anton-l Date: Tue, 31 Aug 2021 17:27:40 +0300 Subject: [PATCH] Upload --- config.json | 91 ++++++++++++++++++++++++++++++++++++++++ preprocessor_config.json | 9 ++++ pytorch_model.bin | 3 ++ 3 files changed, 103 insertions(+) create mode 100644 config.json create mode 100644 preprocessor_config.json create mode 100644 pytorch_model.bin diff --git a/config.json b/config.json new file mode 100644 index 0000000..2b4bd5b --- /dev/null +++ b/config.json @@ -0,0 +1,91 @@ +{ + "_name_or_path": "anton-l/wav2vec2-random-tiny-classifier", + "activation_dropout": 0.0, + "apply_spec_augment": true, + "architectures": [ + "Wav2Vec2ForSequenceClassification" + ], + "attention_dropout": 0.1, + "bos_token_id": 1, + "classifier_proj_size": 16, + "codevector_dim": 256, + "contrastive_logits_temperature": 0.1, + "conv_bias": false, + "conv_dim": [ + 32 + ], + "conv_kernel": [ + 2 + ], + "conv_stride": [ + 320 + ], + "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, + "gradient_checkpointing": true, + "hidden_act": "gelu", + "hidden_dropout": 0.1, + "hidden_size": 32, + "id2label": { + "0": "yes", + "1": "no", + "2": "up", + "3": "down", + "4": "left", + "5": "right", + "6": "on", + "7": "off", + "8": "stop", + "9": "go", + "10": "_unknown_", + "11": "_silence_" + }, + "initializer_range": 0.02, + "intermediate_size": 32, + "label2id": { + "_silence_": 11, + "_unknown_": 10, + "down": 3, + "go": 9, + "left": 4, + "no": 1, + "off": 7, + "on": 6, + "right": 5, + "stop": 8, + "up": 2, + "yes": 0 + }, + "layer_norm_eps": 1e-05, + "layerdrop": 0.05, + "mask_feature_length": 10, + "mask_feature_prob": 0.0, + "mask_time_length": 10, + "mask_time_prob": 0.05, + "model_type": "wav2vec2", + "no_mask_channel_overlap": false, + "no_mask_time_overlap": false, + "num_attention_heads": 2, + "num_codevector_groups": 2, + "num_codevectors_per_group": 320, + "num_conv_pos_embedding_groups": 2, + "num_conv_pos_embeddings": 8, + "num_feat_extract_layers": 1, + "num_hidden_layers": 2, + "num_negatives": 100, + "pad_token_id": 0, + "proj_codevector_dim": 256, + "torch_dtype": "float32", + "transformers_version": "4.10.0.dev0", + "use_weighted_layer_sum": true, + "vocab_size": 32 +} diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..10f6def --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,9 @@ +{ + "do_normalize": false, + "feature_extractor_type": "Wav2Vec2FeatureExtractor", + "feature_size": 1, + "padding_side": "right", + "padding_value": 0.0, + "return_attention_mask": true, + "sampling_rate": 16000 +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..53aac4a --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9644bc4ac6bc52b76ca7342faa1304a68154f398ad1adf08f6cc5545dfc96198 +size 94607