yangwang825 commited on
Commit
dc937a2
1 Parent(s): 9b12b7b

Upload config

Browse files
Files changed (1) hide show
  1. config.json +5 -3
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_name_or_path": "facebook/wav2vec2-base",
3
  "activation_dropout": 0.0,
4
  "adapter_attn_dim": null,
5
  "adapter_kernel_size": 3,
@@ -7,9 +6,12 @@
7
  "add_adapter": false,
8
  "apply_spec_augment": true,
9
  "architectures": [
10
- "Wav2Vec2SpkRegModel"
11
  ],
12
  "attention_dropout": 0.1,
 
 
 
13
  "bos_token_id": 1,
14
  "classifier_proj_size": 256,
15
  "codevector_dim": 256,
@@ -54,6 +56,7 @@
54
  "feat_quantizer_dropout": 0.0,
55
  "final_dropout": 0.0,
56
  "freeze_feat_extract_train": true,
 
57
  "hidden_act": "gelu",
58
  "hidden_dropout": 0.1,
59
  "hidden_size": 768,
@@ -117,7 +120,6 @@
117
  1,
118
  1
119
  ],
120
- "torch_dtype": "float32",
121
  "transformers_version": "4.46.2",
122
  "use_weighted_layer_sum": false,
123
  "vocab_size": 32,
 
1
  {
 
2
  "activation_dropout": 0.0,
3
  "adapter_attn_dim": null,
4
  "adapter_kernel_size": 3,
 
6
  "add_adapter": false,
7
  "apply_spec_augment": true,
8
  "architectures": [
9
+ "Wav2Vec2ForPreTraining"
10
  ],
11
  "attention_dropout": 0.1,
12
+ "auto_map": {
13
+ "AutoConfig": "configuration_wav2vec2_spkreg.Wav2Vec2SpkRegConfig"
14
+ },
15
  "bos_token_id": 1,
16
  "classifier_proj_size": 256,
17
  "codevector_dim": 256,
 
56
  "feat_quantizer_dropout": 0.0,
57
  "final_dropout": 0.0,
58
  "freeze_feat_extract_train": true,
59
+ "gradient_checkpointing": true,
60
  "hidden_act": "gelu",
61
  "hidden_dropout": 0.1,
62
  "hidden_size": 768,
 
120
  1,
121
  1
122
  ],
 
123
  "transformers_version": "4.46.2",
124
  "use_weighted_layer_sum": false,
125
  "vocab_size": 32,