File size: 954 Bytes
7694c84
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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

# restore_model: ''
restore_model: /content/drive/MyDrive/AR-SA-TTS/v1/states.pth
# restore_model: ./checkpoints/exp_tc2_adv/states.pth

log_dir: logs/exp_tc2_adv
checkpoint_dir: /content/drive/MyDrive/AR-SA-TTS/v1

train_wavs_path: /content/modeltts-ar/newdata
train_labels: ./data/SA/train_arab.txt

test_wavs_path: /content/modeltts-ar/newdata
test_labels: ./data/SA/train_phon.txt

label_pattern: '"(?P<filename>.*)" "(?P<arabic>.*)"'
#label_pattern: (?P<arabic>.*)\|(?P<filestem>.*)

# loss weights
gan_loss_weight: 4.
feat_loss_weight: 1.

# optimizers
g_lr: 1.0e-4    # lr for AdamW optimizer (generator)
g_beta1: 0.9     # beta1 for AdamW optimizer (generator)
g_beta2: 0.99   # beta2 for AdamW optimizer (generator)

d_lr: 1.0e-4    # lr for AdamW optimizer (discriminator)
d_beta1: 0.8     # beta1 for AdamW optimizer (discriminator)
d_beta2: 0.99   # beta2 for AdamW optimizer (discriminator)

n_save_states_iter: 10
n_save_backup_iter: 1000