wehub-resource-sync
96 lines
2.0 KiB
YAML
96 lines
2.0 KiB
YAML
# task
|
|
base_config: configs/config_base.yaml
|
|
task_cls: ''
|
|
#############
|
|
# dataset
|
|
#############
|
|
raw_data_dir: ''
|
|
processed_data_dir: ''
|
|
binary_data_dir: ''
|
|
dict_dir: ''
|
|
pre_align_cls: ''
|
|
binarizer_cls: data_gen.tts.base_binarizer.BaseBinarizer
|
|
pre_align_args:
|
|
use_tone: true # for ZH
|
|
forced_align: mfa
|
|
use_sox: false
|
|
txt_processor: en
|
|
allow_no_txt: false
|
|
denoise: false
|
|
binarization_args:
|
|
shuffle: false
|
|
with_txt: true
|
|
with_wav: false
|
|
with_align: true
|
|
with_spk_embed: true
|
|
with_f0: true
|
|
with_f0cwt: true
|
|
|
|
loud_norm: false
|
|
endless_ds: true
|
|
reset_phone_dict: true
|
|
|
|
test_num: 100
|
|
valid_num: 100
|
|
max_frames: 1550
|
|
max_input_tokens: 1550
|
|
audio_num_mel_bins: 80
|
|
audio_sample_rate: 22050
|
|
hop_size: 256 # For 22050Hz, 275 ~= 12.5 ms (0.0125 * sample_rate)
|
|
win_size: 1024 # For 22050Hz, 1100 ~= 50 ms (If None, win_size: fft_size) (0.05 * sample_rate)
|
|
fmin: 80 # Set this to 55 if your speaker is male! if female, 95 should help taking off noise. (To test depending on dataset. Pitch info: male~[65, 260], female~[100, 525])
|
|
fmax: 7600 # To be increased/reduced depending on data.
|
|
fft_size: 1024 # Extra window size is filled with 0 paddings to match this parameter
|
|
min_level_db: -100
|
|
num_spk: 1
|
|
mel_vmin: -6
|
|
mel_vmax: 1.5
|
|
ds_workers: 4
|
|
|
|
#########
|
|
# model
|
|
#########
|
|
dropout: 0.1
|
|
enc_layers: 4
|
|
dec_layers: 4
|
|
hidden_size: 384
|
|
num_heads: 2
|
|
prenet_dropout: 0.5
|
|
prenet_hidden_size: 256
|
|
stop_token_weight: 5.0
|
|
enc_ffn_kernel_size: 9
|
|
dec_ffn_kernel_size: 9
|
|
ffn_act: gelu
|
|
ffn_padding: 'SAME'
|
|
|
|
|
|
###########
|
|
# optimization
|
|
###########
|
|
lr: 2.0
|
|
warmup_updates: 8000
|
|
optimizer_adam_beta1: 0.9
|
|
optimizer_adam_beta2: 0.98
|
|
weight_decay: 0
|
|
clip_grad_norm: 1
|
|
|
|
|
|
###########
|
|
# train and eval
|
|
###########
|
|
max_tokens: 30000
|
|
max_sentences: 100000
|
|
max_eval_sentences: 1
|
|
max_eval_tokens: 60000
|
|
train_set_name: 'train'
|
|
valid_set_name: 'valid'
|
|
test_set_name: 'test'
|
|
vocoder: pwg
|
|
vocoder_ckpt: ''
|
|
profile_infer: false
|
|
out_wav_norm: false
|
|
save_gt: false
|
|
save_f0: false
|
|
gen_dir_name: ''
|
|
use_denoise: false
|