sourabharsh commited on
Commit
6b102fb
·
1 Parent(s): a31d839
Files changed (2) hide show
  1. config.json +19 -10
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "_name_or_path": "facebook/wav2vec2-base-960h",
3
- "activation_dropout": 0.1,
4
  "apply_spec_augment": true,
5
  "architectures": [
6
  "Wav2Vec2ForCTC"
@@ -37,31 +37,40 @@
37
  2,
38
  2
39
  ],
40
- "ctc_loss_reduction": "sum",
41
- "ctc_zero_infinity": false,
42
  "diversity_loss_weight": 0.1,
43
  "do_stable_layer_norm": false,
44
  "eos_token_id": 2,
45
  "feat_extract_activation": "gelu",
46
- "feat_extract_dropout": 0.0,
47
  "feat_extract_norm": "group",
48
  "feat_proj_dropout": 0.1,
49
  "feat_quantizer_dropout": 0.0,
50
- "final_dropout": 0.1,
51
- "gradient_checkpointing": false,
 
52
  "hidden_act": "gelu",
53
  "hidden_dropout": 0.1,
54
- "hidden_dropout_prob": 0.1,
55
  "hidden_size": 768,
56
  "initializer_range": 0.02,
57
  "intermediate_size": 3072,
58
  "layer_norm_eps": 1e-05,
59
- "layerdrop": 0.1,
 
 
 
 
 
60
  "mask_feature_length": 10,
61
  "mask_feature_prob": 0.0,
62
  "mask_time_length": 10,
 
 
63
  "mask_time_prob": 0.05,
 
64
  "model_type": "wav2vec2",
 
 
65
  "num_attention_heads": 12,
66
  "num_codevector_groups": 2,
67
  "num_codevectors_per_group": 320,
@@ -70,7 +79,7 @@
70
  "num_feat_extract_layers": 7,
71
  "num_hidden_layers": 12,
72
  "num_negatives": 100,
73
- "pad_token_id": 0,
74
  "proj_codevector_dim": 256,
75
  "transformers_version": "4.8.2",
76
  "vocab_size": 32
 
1
  {
2
+ "_name_or_path": "output/checkpoint-3600",
3
+ "activation_dropout": 0.0,
4
  "apply_spec_augment": true,
5
  "architectures": [
6
  "Wav2Vec2ForCTC"
 
37
  2,
38
  2
39
  ],
40
+ "ctc_loss_reduction": "mean",
41
+ "ctc_zero_infinity": true,
42
  "diversity_loss_weight": 0.1,
43
  "do_stable_layer_norm": false,
44
  "eos_token_id": 2,
45
  "feat_extract_activation": "gelu",
 
46
  "feat_extract_norm": "group",
47
  "feat_proj_dropout": 0.1,
48
  "feat_quantizer_dropout": 0.0,
49
+ "final_dropout": 0.0,
50
+ "freeze_feat_extract_train": true,
51
+ "gradient_checkpointing": true,
52
  "hidden_act": "gelu",
53
  "hidden_dropout": 0.1,
 
54
  "hidden_size": 768,
55
  "initializer_range": 0.02,
56
  "intermediate_size": 3072,
57
  "layer_norm_eps": 1e-05,
58
+ "layerdrop": 0.05,
59
+ "mask_channel_length": 10,
60
+ "mask_channel_min_space": 1,
61
+ "mask_channel_other": 0.0,
62
+ "mask_channel_prob": 0.0,
63
+ "mask_channel_selection": "static",
64
  "mask_feature_length": 10,
65
  "mask_feature_prob": 0.0,
66
  "mask_time_length": 10,
67
+ "mask_time_min_space": 1,
68
+ "mask_time_other": 0.0,
69
  "mask_time_prob": 0.05,
70
+ "mask_time_selection": "static",
71
  "model_type": "wav2vec2",
72
+ "no_mask_channel_overlap": false,
73
+ "no_mask_time_overlap": false,
74
  "num_attention_heads": 12,
75
  "num_codevector_groups": 2,
76
  "num_codevectors_per_group": 320,
 
79
  "num_feat_extract_layers": 7,
80
  "num_hidden_layers": 12,
81
  "num_negatives": 100,
82
+ "pad_token_id": 29,
83
  "proj_codevector_dim": 256,
84
  "transformers_version": "4.8.2",
85
  "vocab_size": 32
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e98cae16b8b3fffe1637c2a186cf2edbc4ee5ccbf2e04b32bed56176d57b1a7f
3
- size 377667031
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43917d3dfe3c01f7cce6a8f4644aebd16fd7023de0eedc229b4fc3b79566e425
3
+ size 377670039