From 6bb7161231899f8eb8669f65213b28e5769fb43b Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sun, 13 Jun 2021 20:53:32 +0100 Subject: [PATCH] add model --- config.json | 28 +++++++++++++++++++++++++++- tf_model.h5 | 4 ++-- 2 files changed, 29 insertions(+), 3 deletions(-) diff --git a/config.json b/config.json index a1fc909..674493e 100644 --- a/config.json +++ b/config.json @@ -1,7 +1,14 @@ { + "_name_or_path": "facebook/wav2vec2-large-960h-lv60-self", + "activation_dropout": 0.1, + "apply_spec_augment": true, "architectures": [ "Wav2Vec2ForCTC" ], + "attention_dropout": 0.1, + "bos_token_id": 1, + "codevector_dim": 256, + "contrastive_logits_temperature": 0.1, "conv_bias": true, "conv_dim": [ 512, @@ -30,22 +37,41 @@ 2, 2 ], + "ctc_loss_reduction": "sum", + "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.1, + "feat_quantizer_dropout": 0.0, + "final_dropout": 0.1, + "gradient_checkpointing": false, "hidden_act": "gelu", + "hidden_dropout": 0.1, "hidden_dropout_prob": 0.1, "hidden_size": 1024, "initializer_range": 0.02, "intermediate_size": 4096, "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": "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, - "transformers_version": "4.3.0.dev0", + "num_negatives": 100, + "pad_token_id": 0, + "proj_codevector_dim": 256, + "transformers_version": "4.7.0.dev0", "vocab_size": 32 } diff --git a/tf_model.h5 b/tf_model.h5 index 17d601d..162bd8e 100644 --- a/tf_model.h5 +++ b/tf_model.h5 @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:2976f0f42b8cc9c663896263d18316c090f0f19f9f16008c241f692d497a2960 -size 1262392464 +oid sha256:924217bb609535355134d3da00d37c747177c1366f8a5f296bb4822942cb6add +size 1262396960