Delete configs
Browse files- configs/__init__.py +0 -18
- configs/config.py +0 -47
- configs/config.yml +0 -49
configs/__init__.py
DELETED
@@ -1,18 +0,0 @@
|
|
1 |
-
import re
|
2 |
-
import yaml
|
3 |
-
|
4 |
-
|
5 |
-
def load_yaml(path):
|
6 |
-
loader = yaml.SafeLoader
|
7 |
-
loader.add_implicit_resolver(
|
8 |
-
u'tag:yaml.org,2002:float',
|
9 |
-
re.compile(u'''^(?:
|
10 |
-
[-+]?(?:[0-9][0-9_]*)\\.[0-9_]*(?:[eE][-+]?[0-9]+)?
|
11 |
-
|[-+]?(?:[0-9][0-9_]*)(?:[eE][-+]?[0-9]+)
|
12 |
-
|\\.[0-9_]+(?:[eE][-+][0-9]+)?
|
13 |
-
|[-+]?[0-9][0-9_]*(?::[0-5]?[0-9])+\\.[0-9_]*
|
14 |
-
|[-+]?\\.(?:inf|Inf|INF)
|
15 |
-
|\\.(?:nan|NaN|NAN))$''', re.X),
|
16 |
-
list(u'-+0123456789.'))
|
17 |
-
with open(path, "r", encoding="utf-8") as file:
|
18 |
-
return yaml.load(file, Loader=loader)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
configs/config.py
DELETED
@@ -1,47 +0,0 @@
|
|
1 |
-
# Copyright 2023 by zhongying
|
2 |
-
#
|
3 |
-
|
4 |
-
|
5 |
-
from . import load_yaml
|
6 |
-
from util.utils import preprocess_paths
|
7 |
-
|
8 |
-
|
9 |
-
class Config:
|
10 |
-
""" User configs class for training, testing or infering """
|
11 |
-
|
12 |
-
def __init__(self, path: str):
|
13 |
-
print('configs file path:', path)
|
14 |
-
config = load_yaml(preprocess_paths(path))
|
15 |
-
self.speech_config = config.get("speech_config", {})
|
16 |
-
self.model_config = config.get("model_config", {})
|
17 |
-
self.dataset_config = config.get("dataset_config", {})
|
18 |
-
self.optimizer_config = config.get("optimizer_config", {})
|
19 |
-
self.running_config = config.get("running_config", {})
|
20 |
-
|
21 |
-
def print(self):
|
22 |
-
print('==================================================')
|
23 |
-
print('speech configs:', self.speech_config)
|
24 |
-
print('--------------------------------------------------')
|
25 |
-
print('model configs:', self.model_config)
|
26 |
-
print('--------------------------------------------------')
|
27 |
-
print('dataset configs:', self.dataset_config)
|
28 |
-
print('--------------------------------------------------')
|
29 |
-
print('optimizer configs', self.optimizer_config)
|
30 |
-
print('--------------------------------------------------')
|
31 |
-
print('running configs:', self.running_config)
|
32 |
-
print('==================================================')
|
33 |
-
|
34 |
-
def toString(self):
|
35 |
-
string = ''
|
36 |
-
string += '#==================================================' + '\n'
|
37 |
-
string += '#speech config: ' + str(self.speech_config) + '\n'
|
38 |
-
string += '#--------------------------------------------------' + '\n'
|
39 |
-
string += '#model config: ' + str(self.model_config) + '\n'
|
40 |
-
string += '#--------------------------------------------------' + '\n'
|
41 |
-
string += '#dataset config: ' + str(self.dataset_config) + '\n'
|
42 |
-
string += '#--------------------------------------------------' + '\n'
|
43 |
-
string += '#optimizer config: ' + str(self.optimizer_config) + '\n'
|
44 |
-
string += '#--------------------------------------------------' + '\n'
|
45 |
-
string += '#running config: ' + str(self.running_config) + '\n'
|
46 |
-
string += '#==================================================' + '\n'
|
47 |
-
return string
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
configs/config.yml
DELETED
@@ -1,49 +0,0 @@
|
|
1 |
-
speech_config:
|
2 |
-
sample_rate: 16000
|
3 |
-
frame_ms: 25
|
4 |
-
stride_ms: 10
|
5 |
-
num_feature_bins: 80
|
6 |
-
feature_type: log_mel_spectrogram
|
7 |
-
preemphasis: 0.97
|
8 |
-
normalize_signal: True
|
9 |
-
normalize_feature: True
|
10 |
-
normalize_per_feature: False
|
11 |
-
|
12 |
-
model_config:
|
13 |
-
name: acrnn
|
14 |
-
d_model: 64
|
15 |
-
filters: [32,64,64]
|
16 |
-
kernel_size: [[11,5],[11,5],[11,5]]
|
17 |
-
rnn_cell: 256
|
18 |
-
seq_mask: True
|
19 |
-
|
20 |
-
dataset_config:
|
21 |
-
vocabulary: vocab/vocab.txt
|
22 |
-
data_path: ./data/wavs/
|
23 |
-
corpus_name: ./data/demo_txt/demo
|
24 |
-
file_nums: 1
|
25 |
-
max_audio_length: 2000
|
26 |
-
shuffle_size: 1200
|
27 |
-
data_length: None
|
28 |
-
suffix: .txt
|
29 |
-
load_type: txt
|
30 |
-
train: train
|
31 |
-
dev: dev
|
32 |
-
test: test
|
33 |
-
|
34 |
-
optimizer_config:
|
35 |
-
init_steps: 0
|
36 |
-
warmup_steps: 10000
|
37 |
-
max_lr: 1e-4
|
38 |
-
beta1: 0.9
|
39 |
-
beta2: 0.999
|
40 |
-
epsilon: 1e-9
|
41 |
-
|
42 |
-
running_config:
|
43 |
-
prefetch: False
|
44 |
-
load_weights: ./saved_weights/20230228-084356/last/model
|
45 |
-
num_epochs: 100
|
46 |
-
batch_size: 1
|
47 |
-
train_steps: 50
|
48 |
-
dev_steps: 10
|
49 |
-
test_steps: 10
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|