-
Notifications
You must be signed in to change notification settings - Fork 37
/
hyperparams.py
58 lines (52 loc) · 1.1 KB
/
hyperparams.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# Audio
num_mels = 80
n_fft = 2048
sr = 22050
preemphasis = 0.97
hop_length = 256
win_length = 1024
n_mels = 80 # Number of Mel banks to generate
power = 1.2 # Exponent for amplifying the predicted magnitude
min_level_db = -100
ref_level_db = 20
n_layers = 3
n_heads = 4
text_hidden_size = 256
embedding_size = 512
history_hidden_size = 256
hidden_size = 256
max_db = 100
ref_db = 20
num_emo = 5
emo_emb = 64
ref_filter_size1 = 128
ref_filter_size2 = 256
ref_filter_size3 = 128
ref_kernel_size = 5
ref_gru_width = 64
style_size = 4
coarse_size = 32
n_iter = 60
# power = 1.5
outputs_per_step = 1
epochs = 10000
lr = 0.001
save_step = 2000
val_step = 2000
image_step = 500
batch_size = 16
n_gpus = 1
cleaners='english_cleaners'
accum=5
restore_step = 694000
restore_step_post = 0
data_path = '/home/DB_location'
train_data_csv = 'metadata_train.csv'
val_data_csv = 'metadata_val.csv'
test_data_csv = 'metadata_test.csv'
test_data_csv_shuf = 'metadata_ref.csv'
checkpoint_path = './checkpoints'
log_directory = 'transformer-tts-phone'
sample_path = './samples'
preprocess_path = './preprocessed'
stop_epoch = 2000