From 4a779774d9473a62b9436b87cf9ac885b97e6f16 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Wed, 26 Jan 2022 13:47:21 +0000
Subject: [PATCH] add head

---
 config.json       | 31 ++++++++++++++++++++++++++++---
 pytorch_model.bin |  4 ++--
 2 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/config.json b/config.json
index a886a7b..5982492 100644
--- a/config.json
+++ b/config.json
@@ -1,8 +1,9 @@
 {
+  "_name_or_path": "./",
   "activation_dropout": 0.1,
   "apply_spec_augment": true,
   "architectures": [
-    "UniSpeechSatForCTC"
+    "UniSpeechSatForSequenceClassification"
   ],
   "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-sat",
   "num_attention_heads": 2,
@@ -62,8 +65,30 @@
   "pad_token_id": 0,
   "proj_codevector_dim": 256,
   "replace_prob": 0.5,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
   "torch_dtype": "float32",
-  "transformers_version": "4.12.0.dev0",
+  "transformers_version": "4.16.0.dev0",
   "use_weighted_layer_sum": false,
-  "vocab_size": 32
+  "vocab_size": 32,
+  "xvector_output_dim": 512
 }
diff --git a/pytorch_model.bin b/pytorch_model.bin
index 8e3dd60..2b824f8 100644
--- a/pytorch_model.bin
+++ b/pytorch_model.bin
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b23b6f9effbd1924fd270668aa133e69f414d4bb5b4647f2096a37cdfe2ddb49
-size 144684
+oid sha256:25fead51db8fb889d79a235bc72c9962e5306e268c05818fb29b3460e3e909bc
+size 159968