wasm-ara / configs /nawar_fp.yaml
wasmdashai's picture
first commit
4451360
raw
history blame contribute delete
902 Bytes
# restore_model: ''
restore_model: ./pretrained/fastpitch_ar_adv.pth
# restore_model: ./checkpoints/exp_fp/states.pth
log_dir: logs/exp_fp
checkpoint_dir: checkpoints/exp_fp
# dataset
train_wavs_path: G:/data/arabic-speech-corpus/wav_new
train_labels: ./data/train_phon.txt
test_wavs_path: G:/data/arabic-speech-corpus/test set/wav_new
test_labels: ./data/test_phon.txt
label_pattern: '"(?P<filename>.*)" "(?P<phonemes>.*)"'
# label_pattern: (?P<arabic>.*)\|(?P<filestem>.*)
# for fastpitch
f0_dict_path: ./data/pitch_dict.pt
f0_mean: 130.05478
f0_std: 22.86267
# batch sizes
max_lengths: [1000, 1300, 1850, 30000] # 1 frame ≈ 11.6ms
batch_sizes: [10, 8, 6, 4]
# optimizers
g_lr: 1.0e-4 # lr for AdamW optimizer (generator)
g_beta1: 0.9 # beta1 for AdamW optimizer (generator)
g_beta2: 0.999 # beta2 for AdamW optimizer (generator)
n_save_states_iter: 100
n_save_backup_iter: 1000