Upload config
Browse files- config.json +4 -7
config.json
CHANGED
|
@@ -4,14 +4,11 @@
|
|
| 4 |
"activation_function": "gelu",
|
| 5 |
"apply_spec_augment": false,
|
| 6 |
"architectures": [
|
| 7 |
-
"
|
| 8 |
],
|
| 9 |
-
"attention_dropout": 0.
|
| 10 |
"auto_map": {
|
| 11 |
-
"AutoConfig": "configuration_whisper_spkreg.WhisperSpkRegConfig"
|
| 12 |
-
"AutoFeatureExtractor": "feature_extractor_whisper_spkreg.WhisperSpkRegFeatureExtractor",
|
| 13 |
-
"AutoModel": "modeling_whisper_spkreg.WhisperSpkRegModel",
|
| 14 |
-
"AutoModelForAudioClassification": "modeling_whisper_spkreg.WhisperSpkRegForSequenceClassification"
|
| 15 |
},
|
| 16 |
"begin_suppress_tokens": [
|
| 17 |
220,
|
|
@@ -50,7 +47,7 @@
|
|
| 50 |
"is_encoder_decoder": true,
|
| 51 |
"label_smoothing": 0.0,
|
| 52 |
"loss_fct": "cross_entropy",
|
| 53 |
-
"margin": 0.
|
| 54 |
"mask_feature_length": 10,
|
| 55 |
"mask_feature_min_masks": 0,
|
| 56 |
"mask_feature_prob": 0.0,
|
|
|
|
| 4 |
"activation_function": "gelu",
|
| 5 |
"apply_spec_augment": false,
|
| 6 |
"architectures": [
|
| 7 |
+
"WhisperForConditionalGeneration"
|
| 8 |
],
|
| 9 |
+
"attention_dropout": 0.0,
|
| 10 |
"auto_map": {
|
| 11 |
+
"AutoConfig": "configuration_whisper_spkreg.WhisperSpkRegConfig"
|
|
|
|
|
|
|
|
|
|
| 12 |
},
|
| 13 |
"begin_suppress_tokens": [
|
| 14 |
220,
|
|
|
|
| 47 |
"is_encoder_decoder": true,
|
| 48 |
"label_smoothing": 0.0,
|
| 49 |
"loss_fct": "cross_entropy",
|
| 50 |
+
"margin": 0.35,
|
| 51 |
"mask_feature_length": 10,
|
| 52 |
"mask_feature_min_masks": 0,
|
| 53 |
"mask_feature_prob": 0.0,
|