up

config.json

@@ -1,8 +1,9 @@
 {
+  "_name_or_path": "./",
   "activation_dropout": 0.1,
   "apply_spec_augment": true,
   "architectures": [
-    "UniSpeechForPreTraining"
+    "UniSpeechForSequenceClassification"
   ],
   "attention_dropout": 0.1,
   "bos_token_id": 1,
@@ -46,8 +47,10 @@
   "layer_norm_eps": 1e-05,
   "layerdrop": 0.1,
   "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
   "mask_feature_prob": 0.0,
   "mask_time_length": 10,
+  "mask_time_min_masks": 2,
   "mask_time_prob": 0.05,
   "model_type": "unispeech",
   "num_attention_heads": 2,
@@ -63,7 +66,7 @@
   "proj_codevector_dim": 256,
   "replace_prob": 0.5,
   "torch_dtype": "float32",
-  "transformers_version": "4.12.0.dev0",
+  "transformers_version": "4.16.0.dev0",
   "use_weighted_layer_sum": false,
   "vocab_size": 32
 }