SumitMdhr commited on
Commit
ffc80bd
·
verified ·
1 Parent(s): d5263f8

Upload 4 files

Browse files
Files changed (4) hide show
  1. configs.py +28 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +10 -0
  4. 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