Upload 4 files
Browse files- configs.py +28 -0
- model.safetensors +3 -0
- preprocessor_config.json +10 -0
- training_args.bin +3 -0
configs.py
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
from datetime import datetime
|
| 3 |
+
|
| 4 |
+
from mltu.configs import BaseModelConfigs
|
| 5 |
+
|
| 6 |
+
class ModelConfigs(BaseModelConfigs):
|
| 7 |
+
def __init__(self):
|
| 8 |
+
super().__init__()
|
| 9 |
+
self.model_path = os.path.join(
|
| 10 |
+
"Models/10_wav2vec2_torch",
|
| 11 |
+
datetime.strftime(datetime.now(), "%Y%m%d%H%M"),
|
| 12 |
+
)
|
| 13 |
+
self.batch_size = 8
|
| 14 |
+
self.train_epochs = 60
|
| 15 |
+
self.train_workers = 20
|
| 16 |
+
|
| 17 |
+
self.init_lr = 1.0e-8
|
| 18 |
+
self.lr_after_warmup = 1e-05
|
| 19 |
+
self.final_lr = 5e-06
|
| 20 |
+
self.warmup_epochs = 10
|
| 21 |
+
self.decay_epochs = 40
|
| 22 |
+
self.weight_decay = 0.005
|
| 23 |
+
self.mixed_precision = True
|
| 24 |
+
|
| 25 |
+
self.max_audio_length = 246000
|
| 26 |
+
self.max_label_length = 256
|
| 27 |
+
|
| 28 |
+
self.vocab = [' ', "'", 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:47b778bcbee4233a642738ed569fae5ffc542a51f046bd2b9b00fd2ce8d8bf11
|
| 3 |
+
size 1262098580
|
preprocessor_config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"do_normalize": true,
|
| 3 |
+
"feature_extractor_type": "Wav2Vec2FeatureExtractor",
|
| 4 |
+
"feature_size": 1,
|
| 5 |
+
"padding_side": "right",
|
| 6 |
+
"padding_value": 0.0,
|
| 7 |
+
"processor_class": "Wav2Vec2Processor",
|
| 8 |
+
"return_attention_mask": true,
|
| 9 |
+
"sampling_rate": 16000
|
| 10 |
+
}
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:607ecc5d268271534253048cf176fde7032b8dfa58e9b35279e46c4e9da201fb
|
| 3 |
+
size 4600
|